George HotzandGitHub a07caaca0d handle stride 0 variable reshape (#10536)

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

..

codegen

More div conditions (#10432)

2025-05-26 07:36:05 -04:00

engine

the schedule graph is the tensor graph (#10534)

2025-05-27 09:23:57 -07:00

frontend

add onnx frontend stub [pr] (#9558)

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

nn

use float32 for sgd momentum (#10387)

2025-05-17 21:56:44 -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

handle stride 0 variable reshape (#10536)

2025-05-27 10:00:24 -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

fixup typing for @diskcache [pr] (#10529)

2025-05-27 15:30:49 +03: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