4 people b7511cc390 Ordered tc kernel opts (#8426) ...

* ordered tc kernel opts

* hotfix: clean up

* add post_init checks and introduce threads property

* fix amx thread counts

* rename upcast_size to thread_dims

* rename upcast_axes to elements_per_threads

* rename vars

* improve comment

* improve post_init assert comments

* add comment

---------

Co-authored-by: George Hotz <[email protected]>
Co-authored-by: chenyu <[email protected]>

2024-12-29 10:34:54 -05:00

..

__init__.py

Ordered tc kernel opts (#8426)

2024-12-29 10:34:54 -05:00

cstyle.py

Ordered tc kernel opts (#8426)

2024-12-29 10:34:54 -05:00

llvmir.py

typos found by gemini [pr] (#8400)

2024-12-24 22:32:25 -05:00

ptx.py

add back explicit dict[DType, str] in ptx [pr] (#8352)

2024-12-20 13:19:48 -05:00

wgsl.py

more typing work [pr] (#8345)

2024-12-19 21:46:35 -08:00