tinygrad/tinygrad
George Hotz 8743ca40e2
force reduce to be in axis order (#10837)
* force reduce to be in axis order

* disable rule causing loop

* disable that rule

* no ra there

* only move non reduce

* fix tests
2025-06-24 13:00:16 -07:00
..
codegen remove outdated index masking in lowerer [pr] (#10953) 2025-06-24 07:53:30 -07:00
engine add opts_to_apply option to AST KernelInfo (#10950) 2025-06-24 18:55:39 +03:00
frontend onnx parser (#10435) 2025-06-09 12:44:28 -04:00
kernelize force reduce to be in axis order (#10837) 2025-06-24 13:00:16 -07:00
nn optim flatten().shape[0] is numel (#10935) 2025-06-23 13:11:19 -04:00
opt add opts_to_apply option to AST KernelInfo (#10950) 2025-06-24 18:55:39 +03:00
renderer opt transforms the ast into an optimized ast (#10900) 2025-06-22 09:41:26 -07:00
runtime amd: rename dev_iface -> iface to match nv (#10959) 2025-06-24 20:22:19 +03:00
shape hotfix: don't viz the indexing rewrites 2025-06-23 08:20:26 -07:00
uop force reduce to be in axis order (#10837) 2025-06-24 13:00:16 -07:00
viz viz: match canvas fontsize (#10957) 2025-06-24 19:07:06 +03:00
__init__.py move files into uop dir (#10399) 2025-05-18 11:38:28 -07:00
device.py support copyin/copyout/is_allocated for subbuffers (#10869) 2025-06-24 07:49:04 -07:00
dtype.py Add DType.min and DType.max properties (#10749) 2025-06-10 08:31:34 -07:00
gradient.py uops can have multiple metadata (#10479) 2025-05-22 21:35:02 -07:00
helpers.py fix: getenv("NOLOCALS")/NOLOCALS context var (#10927) 2025-06-23 11:23:59 +03:00
py.typed add a single py.typed (#6083) 2024-08-14 17:31:46 -07:00
tensor.py clean up binary_crossentropy_logits (#10958) 2025-06-24 12:23:40 -04:00