George HotzandGitHub f5346d6a1a fix USE_ATOMICS for non float dtypes and make it the default (#14444) ...

* embedded multistep test

* complex test

* with jit

* fix dtypes and reenable USE_ATOMICS

* that test didn't catch anything

2026-01-31 09:44:16 +08:00

..

apps

return types for all math.py function (#14413)

2026-01-28 20:10:11 -05:00

codegen

add few more types [pr] (#14425)

2026-01-29 14:04:09 -05:00

engine

correct var_vals schedule filter (#14451)

2026-01-30 17:10:07 -05:00

mixin

return types for all math.py function (#14413)

2026-01-28 20:10:11 -05:00

nn

fix USE_ATOMICS for non float dtypes and make it the default (#14444)

2026-01-31 09:44:16 +08:00

renderer

return types for all math.py function (#14413)

2026-01-28 20:10:11 -05:00

runtime

device: call free for external_ptr (#14448)

2026-01-30 23:53:17 +03:00

schedule

use atomics for embedding backward (#14400)

2026-01-30 18:10:59 +08:00

uop

prevent UB in long decomp and more emulated tests (#14447)

2026-01-30 19:38:41 -05:00

viz

add call/param UOps (#14433)

2026-01-30 14:51:45 +08:00

__init__.py

move files into uop dir (#10399)

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

device.py

device: call free for external_ptr (#14448)

2026-01-30 23:53:17 +03:00

dtype.py

add InvalidType to ConstType [pr] (#14373)

2026-01-27 14:09:34 -05:00

gradient.py

add call/param UOps (#14433)

2026-01-30 14:51:45 +08:00

helpers.py

use atomics for embedding backward (#14400)

2026-01-30 18:10:59 +08:00

py.typed

add a single py.typed (#6083)

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

tensor.py

use atomics for embedding backward (#14400)

2026-01-30 18:10:59 +08:00