George HotzandGitHub 8325c4f192 tests for multi assign (#10658) ...

* tests for multi assign

* transformer tests

* add that assert

2025-06-05 20:56:40 -07:00

..

codegen

merge process replay and viz captures [pr] (#10581)

2025-06-01 12:30:10 +03:00

engine

remove gbarrier on const (#10656)

2025-06-06 02:36:52 +03: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

AMD_LLVM: support mfma for mi300x (#10625)

2025-06-05 15:55:44 -04:00

runtime

refactor renderer launch bounds [pr] (#10617)

2025-06-05 08:38:04 -07:00

shape

handle stride 0 variable reshape (#10536)

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

uop

tests for multi assign (#10658)

2025-06-05 20:56:40 -07:00

viz

minimal mstack pr to fix allreduce (#10649)

2025-06-05 15:14:53 -07:00

__init__.py

move files into uop dir (#10399)

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

device.py

MSTACK little non-functional changes (#10648)

2025-06-05 13:20:22 -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

merge process replay and viz captures [pr] (#10581)

2025-06-01 12:30:10 +03:00

py.typed

add a single py.typed (#6083)

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

tensor.py

fix failed threefry (#10646)

2025-06-05 17:17:42 -07:00