Commit Graph
7426 Commits
Author SHA1 Message Date
qazal 657822dcdc Merge remote-tracking branch 'origin/no_uop_mutability' into test_rewrite_map 2024-12-30 22:42:09 +02:00
qazal 2a126c145b only the ops changes 2024-12-30 22:41:16 +02:00
qazalandGitHub d157b20027 delete create_schedule, only create_schedule_with_vars [pr] (#8450) 2024-12-31 04:20:53 +08:00
qazalandGitHub 866dfa1f23 create_schedule([x.lazydata]) -> x.schedule() in tests (#8449) 2024-12-31 03:15:52 +08:00
geohot 0addbad36d Happy New Year! Let's get AM merged 2024-12-30 13:15:10 -05:00
George HotzandGitHub aa74b96ccf Merge branch 'master' into no_uop_mutability 2024-12-30 10:58:32 -05:00
George HotzandGitHub 180916257d add children tracking to uop [pr] (#8448) 2024-12-30 10:58:20 -05:00
George HotzandGitHub 993863b77f Merge branch 'master' into no_uop_mutability 2024-12-30 10:06:35 -05:00
geohot 29c14f1cbf hotfix: update tests for no uop mut 2024-12-30 10:05:37 -05:00
George HotzandGitHub 950e41c751 Merge branch 'master' into no_uop_mutability 2024-12-30 10:04:32 -05:00
geohot cd9bab6bca Revert "tensor universe"
This reverts commit c21301852a.
2024-12-30 10:03:25 -05:00
geohot 0419963880 Revert "does this pass test"
This reverts commit ed516b3169.
2024-12-30 10:03:18 -05:00
qazalandGitHub 96f2a83a62 has_st shouldn't exist anymore [pr] (#8446)
* has_st shouldn't exist anymore [pr]

* const also shouldn't be there
2024-12-30 18:52:18 +08:00
qazalandGitHub 7499139239 scheduler renames from the buffer_shape branch [pr] (#8444)
* scheduler refactors and renames from the buffer_shape branch [pr]

* all unmasked sts are allowed here

* only renames
2024-12-30 16:33:38 +08:00
geohot ed516b3169 does this pass test 2024-12-29 19:31:50 -05:00
geohot c21301852a tensor universe 2024-12-29 19:23:49 -05:00
George HotzandGitHub 24a906aa50 Merge branch 'master' into no_uop_mutability 2024-12-29 18:38:43 -05:00
a3c359b28b merge all one op views [pr] (#8412)
* merge all one op views [pr]

* does this work?

* this won't work (yet)

* apply movement ops on top of the BUFFER

* buffer needs to become base next

---------

Co-authored-by: qazal <[email protected]>
Co-authored-by: qazal <[email protected]>
2024-12-30 05:06:49 +08:00
George HotzandGitHub 1d9df4c842 Merge branch 'master' into no_uop_mutability 2024-12-29 14:34:18 -05:00
George HotzandGitHub 0d2400fc7c three assigns to lazydata [pr] (#8443)
* three assigns to lazydata [pr]

* one less

* Revert "one less"

This reverts commit a96268af34.

* that okay

* don't do that
2024-12-29 14:31:00 -05:00
geohot 6d034c71ef fix that test 2024-12-29 13:12:08 -05:00
geohot 040177fac1 fix test failures 2024-12-29 12:55:02 -05:00
geohot ea7f7bf812 fix test_hcq 2024-12-29 12:45:06 -05:00
geohot 300542aef1 err, that 2024-12-29 12:16:29 -05:00
geohot de5c5e8108 unneeded 2024-12-29 12:15:55 -05:00
George HotzandGitHub f9c68e2f03 Merge branch 'master' into no_uop_mutability 2024-12-29 12:14:50 -05:00
George HotzandGitHub b71c51191b tests from remove uop mutability [pr] (#8442)
* tests from remove uop mutability [pr]

* more test fix

* simpler test fix

* remove that
2024-12-29 12:14:10 -05:00
geohot 530343a13f fix test 2024-12-29 12:10:33 -05:00
geohot ff7f15efce them too 2024-12-29 11:56:21 -05:00
geohot 3bfa2e195c lil test fixups 2024-12-29 11:55:24 -05:00
geohot 2a741b61fe more tests pass 2024-12-29 11:45:10 -05:00
geohot 933227199e most tests pass 2024-12-29 11:42:14 -05:00
geohot 44ff1b7424 test fixups 2024-12-29 10:58:35 -05:00
geohot 1cfccc34c1 remove uop mutability [pr] 2024-12-29 10:49:49 -05:00
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
George HotzandGitHub 0c498a7f53 Merge branch 'master' into test_rewrite_map 2024-12-29 10:34:07 -05:00
geohot bcdf403184 child test passes 2024-12-29 10:32:56 -05:00
geohot 14ab43c133 parent test passes 2024-12-29 10:17:56 -05:00
qazalandGitHub 34987a03af const copy folding spec + multi.py behavior [pr] (#8436)
* const copy folding spec + multi behavior [pr]

* copy from clang, move multi test
2024-12-29 23:12:13 +08:00
geohot a2820e91c5 tests that need to pass 2024-12-29 10:01:50 -05:00
George HotzandGitHub dd7aed9dd8 Merge branch 'master' into test_rewrite_map 2024-12-29 09:50:53 -05:00
qazalandGitHub a44cd1e6f7 add collapse_view to the scheduler [pr] (#8440) 2024-12-29 21:30:29 +08:00
qazalandGitHub 98b2854f14 refactor lower_schedule_item to pattern matchers [pr] (#8434) 2024-12-29 17:10:07 +08:00
chenyuandGitHub 0fd6d7482b minor cleanups in state.py [pr] (#8438) 2024-12-28 20:13:25 -05:00
geohot 3c36afda48 test tiny passes 2024-12-28 19:38:59 -05:00
geohot d859a0449f simplier + more tests 2024-12-28 17:08:47 -05:00
geohot aa554d7dac map is only the map 2024-12-28 12:38:14 -05:00
geohot c450830e56 remove uop mutability 2024-12-28 12:35:01 -05:00
geohot a7420fb8f0 oh, frame 2 2024-12-28 11:07:52 -05:00
geohot 54bf874c3e fix gc issue 2024-12-28 10:52:31 -05:00