George Hotz ca19eb3e82 where fold try 2 (#3748) ...

* where fold try 2

* assign fold

* test_where_fold works

* add gated store support to ops_python

---------

Co-authored-by: nimlgen <138685161+nimlgen@users.noreply.github.com>

2024-03-15 07:46:26 -07:00

..

codegen

where fold try 2 (#3748)

2024-03-15 07:46:26 -07:00

features

Fix kwargs in JIT (#3730)

2024-03-14 23:55:19 -04:00

nn

add Tensor.replace (#3738)

2024-03-14 13:34:14 -07:00

renderer

fix broken loops in llvm (#3751)

2024-03-15 11:57:51 +03:00

runtime

where fold try 2 (#3748)

2024-03-15 07:46:26 -07:00

shape

fix st.real_size can be nagative if valid is always false (#3708)

2024-03-12 20:34:07 -04:00

__init__.py

move graph.py and jit.py into features (#3376)

2024-02-12 17:34:34 +01:00

device.py

call self.nbytes (#3736)

2024-03-14 08:10:12 -07:00

dtype.py

Rename .sz to .count on DType (#3413)

2024-02-15 15:03:49 +01:00

helpers.py

working to improve ptx (#3647)

2024-03-07 12:39:31 -08:00

lazy.py

simple LoadOps.ASSIGN (#3745)

2024-03-14 20:44:34 -07:00

mlops.py

reciprocal mlop (#3694)

2024-03-11 16:08:46 -07:00

ops.py

simple LoadOps.ASSIGN (#3745)

2024-03-14 20:44:34 -07:00

realize.py

where fold try 2 (#3748)

2024-03-15 07:46:26 -07:00

tensor.py

simple LoadOps.ASSIGN (#3745)

2024-03-14 20:44:34 -07:00