Commit Graph
9967 Commits
Author SHA1 Message Date
George HotzandGitHub 03fb0c9ad0 Merge branch 'master' into simpler_postrange 2025-08-26 12:06:18 -07:00
George HotzandGitHub d4154e0349 split devectorizing of buf/index (#11855) 2025-08-26 12:05:48 -07:00
geohot f0f7437385 cleanups 2025-08-26 12:02:14 -07:00
George HotzandGitHub 15886fd513 Merge branch 'master' into simpler_postrange 2025-08-26 11:57:18 -07:00
George HotzandGitHub b268755d51 small changes from postopt (#11854) 2025-08-26 11:56:16 -07:00
geohot 4a8d24f69b work 2025-08-26 11:30:44 -07:00
geohot 0c696e64ef work 2025-08-26 11:21:44 -07:00
geohot b0698865b8 bufferize 2025-08-26 10:50:06 -07:00
Sieds LyklesandGitHub a3aeef45cc associative variation of where branch-merging (#11851)
* add rule and test

* change comment
2025-08-26 19:27:05 +02:00
chenyuandGitHub aabe7756be fix type in fold_bitcast [pr] (#11853) 2025-08-26 13:22:30 -04:00
Jordan ChalupkaandGitHub 4785cd959a [TYPED=1] cvar should allow dtype as a tuple (#11770)
* cvar dtype:DType|tuple[DType, ...]|None=None

* fmt

* add a test

* list typeguard as a dep for CI

* extra step to install mypy

* fix venv

* ci fixes

* mv typeguard to testing install group

* simpler TYPED=1 test

* add typeguard to lint group
2025-08-26 12:49:51 -04:00
qazalandGitHub b111076301 viz: fixup click on overlay rect (#11850) 2025-08-26 19:25:42 +03:00
1dd613cb89 test float_to_bf16 round-to-even behavior (#11849)
Co-authored-by: b1tg <[email protected]>
2025-08-26 12:16:10 -04:00
409399c609 fix nan in float_to_bf16 (#11843)
Co-authored-by: b1tg <[email protected]>
2025-08-26 11:42:25 -04:00
qazalandGitHub 43d5d66d34 viz: add UOp ports to edges (#11847)
* viz: add UOp ports to edges

* one edge label

* g.tag styling

* replace with NodeList
2025-08-26 18:31:52 +03:00
chenyuandGitHub f28f613f85 improved float_to_bf16 (#11848)
round instead of truncate
2025-08-26 11:14:06 -04:00
nimlgenandGitHub afe14ccbfa amd: aql default when several xccs (#11832) 2025-08-26 15:16:36 +03:00
qazalandGitHub 3674c0754e viz: small uop click changes (#11846)
* also highlight self

* can always unselect by clicking outside

* less layout
2025-08-26 14:56:13 +03:00
qazalandGitHub f2a3c27372 viz: g.edges() once (#11845) 2025-08-26 13:29:59 +03:00
qazalandGitHub b0df3e62a8 viz: light up srcs and paths on UOp click (#11844)
* viz: light up srcs and paths on UOp click

* safari doesn't have context-stroke

* safari also has a bug

* safari acceptance
2025-08-26 09:03:09 +03:00
qazalandGitHub 6236749867 viz: move rect styles to classes (#11842)
* viz: move rect styles to classes

* add rect
2025-08-26 07:55:34 +03:00
qazalandGitHub 81ffa07439 viz: pass through nodes without a link (#11841) 2025-08-26 07:00:43 +03:00
geohot 4ea015331e sort of wmma 2025-08-25 20:42:06 -07:00
geohot 7a95d19e7e sort of wmma 2025-08-25 20:35:55 -07:00
Sieds LyklesandGitHub 265d287615 add decomp for !x&!y -> !(x|y) (#11836) 2025-08-26 05:21:06 +02:00
chenyuandGitHub 337e979a59 call dtypes.as_const in Tensor(list) (#11840) 2025-08-25 22:08:26 -04:00
George HotzandGitHub c26ba58e4e Merge branch 'master' into simpler_postrange 2025-08-25 18:03:52 -07:00
George HotzandGitHub 215818379b new (post) group for reduce (#11837)
* new (post) group for reduce

* fixes

* leave if

* fix locals

* size

* no vectorized buf

* image fixes

* don't track that

* fix ptx

* name buffer with reduce range

* remove unused in lowerer

* yay DEFINE_REG refactor
2025-08-25 18:03:00 -07:00
chenyuandGitHub ac3449b0c8 truncate_fp16 cleanup (#11838)
native `@` is default
2025-08-25 19:03:41 -04:00
geohot fec335d4c8 fix gfr 2025-08-25 14:54:35 -07:00
geohot 15dac61988 fix locals 2025-08-25 13:48:11 -07:00
geohot 5c7b638120 mergable? 2025-08-25 13:33:56 -07:00
geohot ce1670fd3d add simplify_merge_adj 2025-08-25 13:28:32 -07:00
George HotzandGitHub 6b1e5568dc Merge branch 'master' into simpler_postrange 2025-08-25 11:32:54 -07:00
qazalandGitHub e146418f65 hotfix: profiler content-type is application/octet-stream (#11831) 2025-08-25 15:56:42 +03:00
qazalandGitHub a1f6823060 viz: memory layout in client side (#11830)
* viz: memory layout in client side

* update test_viz
2025-08-25 14:49:33 +03:00
geohot cc62ad4a21 fa 2025-08-24 17:46:02 -07:00
George HotzandGitHub ca739844fc Merge branch 'master' into simpler_postrange 2025-08-24 17:37:48 -07:00
George HotzandGitHub a6dbb09058 changes for postrange (#11828) 2025-08-24 17:37:07 -07:00
George HotzandGitHub a711834380 Merge branch 'master' into simpler_postrange 2025-08-24 17:10:41 -07:00
geohot 0fba33d646 cleaner 2025-08-24 17:09:23 -07:00
geohot 6da57e96a8 flip axis 2025-08-24 16:53:31 -07:00
geohot 1037defce5 working WMMA 2025-08-24 16:46:48 -07:00
geohot 1e6b70568b working WMMA 2025-08-24 16:42:28 -07:00
geohot ffd0329526 optional postopt 2025-08-24 14:42:31 -07:00
geohot 283d39ad3f simpler postrange works 2025-08-24 14:32:36 -07:00
geohot 3c0aa9e488 simpler postrange 2025-08-24 14:04:19 -07:00
George HotzandGitHub 27701ef823 add locals support to rangeify (#11826) 2025-08-24 14:03:12 -07:00
Sieds LyklesandGitHub a286a1a6f7 Fast idiv try removing factors of two before cast (#11824)
* try removing factors of two

* dont return if None

* add test
2025-08-24 20:04:25 +02:00
geohot a03b930339 hotfix: green v2 in docs 2025-08-24 10:25:14 -07:00