geohot
9759899cb7
actually make the file readable
2025-10-08 19:14:02 +08:00
George Hotz and GitHub
29509a1a57
Merge branch 'master' into rtoposort
2025-10-08 18:48:17 +08:00
geohot
8c3242bb45
disable failing test
2025-10-08 18:47:27 +08:00
geohot
b264772ae4
touch up tests + add comments
2025-10-08 18:38:48 +08:00
chenyu and GitHub
ee0382ad99
remove ShapeTracker.invert ( #12520 )
2025-10-08 18:37:34 +08:00
chenyu and GitHub
d5058427ea
remove ShapeTracker.real_size ( #12519 )
2025-10-08 06:15:29 -04:00
geohot
fc01a7cc15
tests
2025-10-08 18:02:58 +08:00
qazal and GitHub
6f26603f06
delete swizzler.py ( #12518 )
...
* delete swizzler
* remove merge_views tests
* don't need rewrites_for_views
* apply_rewrites
2025-10-08 13:02:34 +03:00
geohot
f09f46d028
put in rangeify map
2025-10-08 17:32:40 +08:00
qazal and GitHub
7e0b14243e
delete grouper and kernelize ( #12517 )
...
* delete grouper and kernelize
* +sys.setrecursionlimit
2025-10-08 12:27:26 +03:00
chenyu and GitHub
e701106a64
remove FUSE_ARANGE ( #12511 )
...
it was the default already
2025-10-08 04:54:07 -04:00
qazal and GitHub
ad49f8148b
switch process_replay to rangeify ( #12509 )
2025-10-08 11:26:43 +03:00
George Hotz and GitHub
b4eb5fcac5
Merge branch 'master' into rtoposort
2025-10-08 15:14:43 +08:00
nimlgen and GitHub
4a756a37d8
amd: support rocm7 ( #12502 )
...
* amd: support rocm7
* mock
2025-10-08 14:30:39 +08:00
qazal and GitHub
60b6dca5ba
update some tests instead of expect_rangeify_fails ( #12500 )
...
* update test_clone_doesnt_dedup to use base
* new_flat_buffer passes
* fix test_reorder_expand
* remove the view stuff
* remove that test, we don't want this view const behavior
* test_setitem_becomes_subbuffer is good
2025-10-08 07:42:31 +03:00
qazal and GitHub
84597ed53c
early assert for device mistmatched asts in rangeify ( #12499 )
...
* early assert for device mistmatched asts in rangeify
* alt also passes
2025-10-08 07:19:36 +03:00
qazal and GitHub
2e19354c1c
viz: reorder timeline graphs ( #12498 )
...
* viz: reorder timeline graphs
* update test_viz with the new order
2025-10-08 07:10:23 +03:00
qazal and GitHub
a7cb80bfab
use recursive_property in UOp device ( #12477 )
...
* simple failing test with RecursionError
* switch to @recursive_property
* merge 2
* diff
2025-10-08 06:15:05 +03:00
geohot
d23324779c
rtoposort is fast, can replace rangeify with this
2025-10-08 11:06:12 +08:00
Sieds Lykles and GitHub
b465c17b56
Revert "UOp.factor and add chain sorting ( #12413 )" ( #12492 )
...
This reverts commit e74be4a140 .
2025-10-08 03:20:23 +02:00
George Hotz and GitHub
945cc46475
delete children tracking from uop ( #12491 )
...
* delete children tracking from uop
* uop children no longer exists
* no tracked children
* that test is flaky too
2025-10-08 09:04:14 +08:00
George Hotz and GitHub
12c4963489
add more rangeify pm tests ( #12488 )
2025-10-07 05:45:38 -04:00
George Hotz and GitHub
403fdfcfd4
check spec in test, cleanup vectorize render ( #12484 )
2025-10-07 17:05:50 +08:00
qazal and GitHub
22674798df
assert correctness in test_permuted_assignment [pr] ( #12483 )
2025-10-07 11:42:22 +03:00
George Hotz and GitHub
75ce11593c
test_reshape_match should match ( #12479 )
2025-10-07 16:07:21 +08:00
George Hotz and GitHub
ea7672931f
fix test_matmul_relu_cat ( #12478 )
2025-10-07 02:32:23 -04:00
chenyu and GitHub
7b48f3cc45
failed test case repro for openpilot model ( #12475 )
...
* failed test case repro for openpilot model
* assertEqual
2025-10-07 13:46:43 +08:00
chenyu and GitHub
a5484b767e
remove skipping cast in simplify_valid [pr] ( #12472 )
...
* remove skipping cast in simplify_valid [pr]
unsupported statements are handled in uop_given_valid already. the test failed because (100%x) somehow got simplified
* better test
2025-10-07 00:10:04 -04:00
George Hotz and GitHub
0f25b4b289
move frontend dir to nn [pr] ( #12470 )
2025-10-07 10:42:22 +08:00
qazal and GitHub
f664bcc8bd
use recursive_property in UOp tracing ( #12469 )
...
* test
* simple passing
2025-10-06 21:10:52 +03:00
qazal and GitHub
76e8a3250c
rangeify: late zero folding ( #12464 )
...
* rangeify: late zero folding
* early
* not kernels
* none
* multi
* linter
* mstack is sink comment
* more comment
2025-10-06 12:52:33 +03:00
chenyu and GitHub
a1881b0c17
update test_chicken ( #12466 )
...
logits are close, just numerical
2025-10-06 03:58:44 -04:00
qazal and GitHub
1b1978b9c0
early copy fixup ( #12463 )
...
* simple failing test
* early copy fixup
2025-10-06 06:38:29 +03:00
chenyu and GitHub
c1e85f699c
multi test case for sharded ring allreduce ( #12462 )
...
* multi test case for sharded ring allreduce
triggers `children not making progress` with RANGEIFY
* expect_rangeify_fails
2025-10-05 23:18:24 -04:00
George Hotz and GitHub
46e8ea15c1
split pm_substitute_recurse ( #12460 )
2025-10-05 21:35:50 -04:00
qazal and GitHub
6ad9a688ed
add failing test after "pend substitutes for speed" ( #12457 )
...
* add failing substitute test
* expect_rangeify_fails
2025-10-05 16:10:04 +03:00
qazal and GitHub
4b60121498
fix bmnist torch with RANGEIFY=1 ( #12442 )
...
* fix bmnist torch with RANGEIFY=1
* alt
* test and comment
* this was always wrong
* simple failing test for rangeify
* simple upat to match the old behavior
2025-10-05 12:34:27 +03:00
George Hotz and GitHub
b5f31d7505
earlier seen children ( #12451 )
2025-10-05 15:55:13 +08:00
qazal and GitHub
865d5796f8
add a test for untested Tensor.assign behavior ( #12448 )
...
* add a test for untested Tensor.assign behavior
* better
2025-10-04 12:44:56 +03:00
Sieds Lykles and GitHub
e74be4a140
UOp.factor and add chain sorting ( #12413 )
...
* add ordering
* fix some tests
* fix more tests
* shorten comment
* update test
* add rule and test
* add rule and test
* remove check
* use fold_divmod_congruence instead of simplify
* adjust tests
* shorten line
* new algo
* add test
* add function to un-nest the div
* add UOp.factor
* test UOp.factor
* uop_given_valid tries to factor simplex expression
* shorten line
* symbolic_flat is back
* change that back
* fix those new tests
* new rule for ordering
* factor multiple factors
* no symbolic_flat
* symbolic_flat to there
* move that back
* fix imports
* merge correctly
* linter happy
* add rule
* add a test
* cleanup
* revert that for now
* UOp.factor returns self instead of None
* try all_candidates
* remove or_else
* post index symbolic
* add test
* maket this closer to the original
* increase mac hlb_cifar min step time
* add some ordering tests
* cleanup
* increase pytest timeout time
* check dtype
2025-10-04 06:05:38 +02:00
Sieds Lykles and GitHub
394dc24110
post index symbolic ( #12446 )
...
* post index symbolic
* add test
2025-10-03 23:23:03 +02:00
chenyu and GitHub
9f2b69b870
enable few tests for PTX test_dtype ( #12445 )
2025-10-03 08:56:30 -04:00
chenyu and GitHub
b087663c35
RANGEIFY test_bert uses more ran somehow ( #12443 )
2025-10-03 04:38:53 -04:00
chenyu and GitHub
940a8d5ba9
default IGNORE_OOB=1 ( #12441 )
...
* default IGNORE_OOB=1
z3 can get very slow with RANGEIFY, also update some kernel numbers to what it is
* add to test
2025-10-03 04:16:19 -04:00
hooved and GitHub
1e8945a28c
Training loop for Stable Diffusion mlperf ( #12315 )
...
* add diff
* fix edit error
* match master
* point reference to specific commit
* simplify wandb logging
* remove lr test, dehardcode device
* increase stack size limit
2025-10-03 02:45:38 -04:00
George Hotz and GitHub
c7849ac593
fix test lil model ( #12437 )
...
* fix test lil model
* 4 not 3
2025-10-03 02:28:37 -04:00
Sieds Lykles and GitHub
0047bcc535
undo loaded comparison swap ( #12436 )
...
* add rule
* add a test
2025-10-03 06:57:29 +02:00
chenyu and GitHub
f203d8b221
update RANGEIFY kernel count and test_masked_select ( #12435 )
2025-10-03 00:41:34 -04:00
wozeparrot and GitHub
a6dd5a224b
skip webgpu tests ( #12433 )
2025-10-02 21:31:07 -07:00
chenyu and GitHub
bf99de7b1e
update a few more tests for RANGEIFY ( #12434 )
2025-10-03 00:16:58 -04:00