George HotzandGitHub ee12e801a3 optional fused optimizers (#10549) ...

* enumerate cases of Tensors in the JIT

* optional fused optimizers

* add fused optimizer test

* move that there

* ugh

2025-05-28 13:50:30 -07:00

..

codegen

s/src[0]/buf in lowerer.py [pr] (#10539)

2025-05-27 21:08:54 +03:00

engine

remove becomes map (#10533)

2025-05-27 11:47:11 -07:00

frontend

add onnx frontend stub [pr] (#9558)

2025-03-24 12:24:34 +08:00

nn

optional fused optimizers (#10549)

2025-05-28 13:50:30 -07:00

renderer

align 16 in ptx, metal, cuda and amd (#10450)

2025-05-21 14:38:54 -07:00

runtime

Add method to return field masks for AMDReg (#10511)

2025-05-25 14:47:20 -07:00

shape

handle stride 0 variable reshape (#10536)

2025-05-27 10:00:24 -07:00

uop

remove becomes map (#10533)

2025-05-27 11:47:11 -07:00

viz

fix display of kernel args in viz [pr] (#10502)

2025-05-24 14:09:52 +03:00

__init__.py

move files into uop dir (#10399)

2025-05-18 11:38:28 -07:00

device.py

move files into uop dir (#10399)

2025-05-18 11:38:28 -07:00

dtype.py

don't use real_size for mem_bytes [pr] (#10147)

2025-05-03 09:41:21 -04:00

gradient.py

uops can have multiple metadata (#10479)

2025-05-22 21:35:02 -07:00

helpers.py

optional fused optimizers (#10549)

2025-05-28 13:50:30 -07:00

py.typed

add a single py.typed (#6083)

2024-08-14 17:31:46 -07:00

tensor.py

view gradient is tricky (#10528)

2025-05-26 22:28:30 -07:00