Commit Graph
8321 Commits
Author SHA1 Message Date
George HotzandGitHub 60cbfe4222 Merge branch 'master' into dsp_search 2025-03-26 10:50:00 +08:00
uuuvnandGitHub 2c32126fc8 am: AMRegister refactor (#9572) 2025-03-26 00:52:40 +07:00
chenyuandGitHub cddd750d68 add a failed test case for jit/nojit rand [pr] (#9574)
currently adding jit produced different rand values
2025-03-25 13:32:44 -04:00
nimlgenandGitHub 4cf2b68ca8 am_smi: fix init for newer versions (#9559) 2025-03-25 23:48:05 +07:00
qazalandGitHub a6a5c0aec5 add NULL=1 backend (#9573)
* add NULL=1 backend

* NullAllocator

* line

* metadata should still work

* it shouldn't have memory usage

* Revert "it shouldn't have memory usage"

This reverts commit a9080fdd43.

* back

* null flops
2025-03-25 22:20:52 +08:00
geohot 311df3ff21 fixes 2025-03-25 19:14:57 +08:00
geohot f6e64a5e8e optional conv 2025-03-25 19:00:48 +08:00
geohot 622ff115a3 back 2025-03-25 18:47:40 +08:00
geohot 5a6e8ee268 fix test 2025-03-25 18:37:43 +08:00
geohot a9f1227625 4 faster 2025-03-25 18:32:27 +08:00
geohot 74c2587ef4 4 -> 8 2025-03-25 18:31:43 +08:00
geohot bce252e0b8 devec 0 2025-03-25 17:12:43 +08:00
geohot 66a90a3c92 ugh, fast 2? 2025-03-25 17:08:35 +08:00
qazalandGitHub b60d9976b4 better yaxis formatting in viz memory graph (#9570)
* better bytes format

* pluralize

* 1 less line
2025-03-25 16:50:22 +08:00
geohot 0d76b0d461 acc2 2025-03-25 15:33:38 +08:00
geohot 5e4505d363 kernel 2 54 GFLOPS 2025-03-25 14:04:03 +08:00
geohot 29920b74d5 unsafe disable on device 2025-03-25 13:57:57 +08:00
geohot ccd18a803c faster? 2025-03-25 13:41:34 +08:00
geohot 943bde47ab fast k26 2025-03-25 13:32:11 +08:00
qazalandGitHub faf3b5b245 display kernel metadata in memory viz (#9569)
* display kernel metadata in memory viz

* fix that
2025-03-25 13:14:54 +08:00
qazalandGitHub 52301fe68e move Buffer refcount increment out of schedule.py (#9564)
* move Buffer refcount increment out of schedule.py

* add TestGC.test_assign_refcount

* refcount refers to Ops.BUFFER UOps
2025-03-25 12:08:27 +08:00
geohot 0d10c7ae2f working on kernel 15 2025-03-25 12:00:16 +08:00
qazalandGitHub 262f5a2bd3 hotfix: replace link in viz/readme (#9568) 2025-03-25 10:24:49 +08:00
chenyuandGitHub 6427272bf6 minor update to rand [pr] (#9566) 2025-03-24 18:49:50 -04:00
chenyuandGitHub b0e070e737 remove MOCKGPU workaround in rand (#9565)
also `requires_grad_` to save a line
2025-03-24 17:49:45 -04:00
qazalandGitHub d7c754ce49 failing test for UOp buffer ref count (#9563)
* failing test for UOp buffer ref count

* lint
2025-03-25 00:10:48 +08:00
f90001e1a6 amd llvm render (no_comgr prereq) (#9543)
* amd llvm render

* skip test_div_rounding_mode

---------

Co-authored-by: b1tg <[email protected]>
2025-03-24 22:50:51 +08:00
Priyank PatelandGitHub 4f5e03bd60 better fix inplace detach (#9557) 2025-03-24 22:50:28 +08:00
qazalandGitHub 1c40873962 show buffer info in memory viz (#9562) 2025-03-24 22:12:30 +08:00
qazalandGitHub efaee75656 start viz of memory usage (#9561)
* start viz of memory usage

* polygons/bars + use d3
2025-03-24 19:05:35 +08:00
qazalandGitHub 1cfe6d02fe refactor uop_to_json to return a dict [pr] (#9560) 2025-03-24 16:38:17 +08:00
geohot 3cab6a3d4a 3x3 2025-03-24 16:36:53 +08:00
geohot 22a56cbaea something for 8 2025-03-24 16:16:59 +08:00
geohot afd61730b4 kernel 5 2025-03-24 16:10:08 +08:00
geohot 536556434b padding ish 2025-03-24 15:57:03 +08:00
geohot 52bff5f39d more 2025-03-24 15:33:48 +08:00
geohot 64d0f14d3d broken 2025-03-24 15:21:43 +08:00
nimlgenandGitHub edf9e1bf8d am: move out soc21 to a sep module (#9551)
* am: soc module is not part of am

* am: soc module is not part of am
2025-03-24 14:17:42 +07:00
geohot 1b61cc6ec3 unaligned 2025-03-24 15:13:57 +08:00
geohot 6f792e8045 vmemu 2025-03-24 15:02:40 +08:00
geohot b1f8018bf4 unaligned load 2025-03-24 14:54:11 +08:00
geohot 2eb9241329 better conv 2025-03-24 13:07:14 +08:00
George HotzandGitHub 554a490751 Merge branch 'master' into dsp_search 2025-03-24 12:29:22 +08:00
George HotzandGitHub 74d98eafb8 add onnx frontend stub [pr] (#9558) 2025-03-24 12:24:34 +08:00
geohot de7d6cec3a hotfix: DEBUG 5 prints the ast 2025-03-24 11:43:11 +08:00
geohot 651c678edf work 2025-03-24 09:49:53 +08:00
chenyuandGitHub ba41076e94 update embedding test to not use dtypes.long [pr] (#9556) 2025-03-23 21:33:38 -04:00
chenyuandGitHub c965f4c20b update bert config (#9555)
BEAM 4->5 for green, 2% faster
use AMD driver instead of AM for red, 5% faster
2025-03-23 16:14:41 -04:00
chenyuandGitHub d734e24c01 minor WEBGPU_PATH cleanup [pr] (#9552)
also mypy recognizes `sys.platform == 'win32'` but does not recognizes it if wrapped inside a helper...
2025-03-23 09:10:02 -04:00
Ahmed HarmoucheandGitHub 7ce7fe0574 Refactor webgpu_dawn lib finding (#9547)
* Refactor webgpu_dawn lib finding

* Fix ruff
2025-03-23 08:23:29 -04:00