| .. |
|
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 |