sirhcm/tinygrad
1
0
Fork 0
forked from tinygrad/tinygrad
Code Pull Requests 2 Actions Activity
6,397 Commits 336 Branches 15 Tags
ea2efbf508b005267de388bd7fad3f2a63672ce2
Commit Graph
6397 Commits
This Branch
This Branch
All Branches
Author SHA1 Message Date
chenyuandGitHub ea2efbf508 Add Opt(op=OptOps.LOCAL, axis=6, amt=2) to actions (#7147) 
* Add Opt(op=OptOps.LOCAL, axis=6, amt=2) to actions

it's missing if we rebuild all kernels, not just the first 2k.

```
PYTHONPATH="." GPU=1 python3 extra/optimization/get_action_space.py
 29%|█████████████████████████████████████▋                                                                                            | 3682/12701 [01:42<04:11, 35.83it/s]Traceback (most recent call last):
  File "/Users/chenyu/code/tinygrad/extra/optimization/get_action_space.py", line 27, in <module>
    test_rebuild(lin)
  File "/Users/chenyu/code/tinygrad/extra/optimization/get_action_space.py", line 11, in test_rebuild
    assert o in actions, f"{o} is not in actions"
           ^^^^^^^^^^^^
AssertionError: Opt(op=OptOps.LOCAL, axis=6, amt=2) is not in actions
```

* break
 2024-10-18 09:03:24 -04:00
qazalandGitHub 4cf7cca91a delete fuzz_schedule [pr] (#7144) 2024-10-18 15:09:39 +03:00
Bhavya GadaandGitHub b7b2017cb9 only ignore warnings not errors (#7146) 2024-10-18 07:41:11 -04:00
ignaciosicaandGitHub 8bcdd7c97d Refactor AMD pm rules to remove handwritten bf16 bool alus (#7136) 
* refactor pm rules

- remove unused handwritten methods
- refactor amd pm rules to fix bug with bool alu

* add bf16 bool alu tests

* add bf16 tests

* hotfix: make atol consistent
 2024-10-18 09:00:46 +08:00
Bhavya GadaandGitHub 534597e753 fix all test warnings (#7024) 
* fix pytorch warning in nn.conv2d for same padding

* fix future warning in torch load

* fix overflow warning in tensor list test: https://github.com/numpy/numpy/issues/23606#issuecomment-1512752172

* fix floating point warnings in dtype tests using docs https://numpy.org/doc/stable/reference/generated/numpy.errstate.html and a neat solution https://stackoverflow.com/questions/53634965/change-np-seterr-behavior-inside-a-function-only

* put err state in one place; comment taken care of by function hover

* enter np errstate context manager on test setup

* put decorator on class
 2024-10-18 08:56:40 +08:00
chenyuandGitHub 0cd4b93441 remove CStyleLanguage from test_uop_symbolic (#7142) 2024-10-17 19:39:34 -04:00
chenyuandGitHub 72ed66205d enable test_resnet_half (#7141) 
already worked so just fixed the test
 2024-10-17 19:02:20 -04:00
nimlgenandGitHub 211d9753f8 nv more lc checks (#7139) 
* nv more lc checks

* revert

* linter
 2024-10-18 00:21:53 +03:00
chenyuandGitHub 12ff52b88b test_failure_52 fails on real METAL (#7138) 2024-10-17 15:37:28 -04:00
chenyuandGitHub 84e98900e8 test linearizer failure 53 (#7137) 
variable scope issue caused compile error
 2024-10-17 15:23:43 -04:00
qazalandGitHub a64e5d0430 graph rewrite all metaops (#7134) 2024-10-17 18:49:20 +03:00
nimlgenandGitHub 45db7d9045 fuzz qcom vs opencl (#7130) 
* fuzz qcom vs opencl

* fix nv

* bettre?

* typo

* open both devs
 2024-10-17 18:49:08 +03:00
qazalandGitHub 188eef959d early rewrite UOps.CONTIGUOUS (#7132) 
* early rewrite UOps.CONTIGUOUS

* add metaops too

* just the contig diff
 2024-10-17 18:35:19 +03:00
chenyuandGitHub 287a198c4f increase test_strongly_connected_DAG threshold (#7131) 
flaky
 2024-10-17 11:08:50 -04:00
George HotzandGitHub c23ef7e2f8 real_remove_const (#7128) 2024-10-17 21:58:41 +08:00
qazalandGitHub 2087abc999 get membufs with dedup [pr] (#7127) 2024-10-17 16:06:06 +03:00
George HotzandGitHub be9a433a60 fix a bug in flops counting + touchups [pr] (#7126) 2024-10-17 21:02:11 +08:00
qazalandGitHub a2eefa6f97 move assign st override to upat (#7122) 
* move assign st override to upat

* merge view
 2024-10-17 13:33:37 +03:00
George HotzandGitHub ded1b38b84 minor dtype cleanup [pr] (#7124) 
* minor dtype cleanup [pr]

* use ptr() function
 2024-10-17 17:41:23 +08:00
George HotzandGitHub 0b2621f63f improve render_dtype [pr] (#7117) 
* improve render_dtype [pr]

* don't deref in index
 2024-10-17 14:50:40 +08:00
George HotzandGitHub ca0dca35f7 move ptx renderer [pr] (#7118) 2024-10-17 14:50:32 +08:00
George HotzandGitHub d990a16326 fix tests to use render (#7116) 2024-10-17 14:35:22 +08:00
geohot 9f4ca88218 hotfix: relax target pct for beautiful_mnist 2024-10-17 12:36:07 +08:00
chenyuandGitHub 51cd0e7c0d idx_given_valid -> uop_given_valid [pr] (#7110) 
will reuse this to simplify valid independent of idx
 2024-10-16 18:16:36 -04:00
chenyuandGitHub 842fe444df test case for valid only simplification (#7108) 2024-10-16 16:40:46 -04:00
chenyuandGitHub 9d109c5382 remove outdated symbolic comments (#7105) 2024-10-16 14:51:59 -04:00
Francis LataGitHubchenyu
 		90eff347e2 tinytqdm write support (#6359) 
* add write support

* add test

* update test case to compare write outputs

* assert final write output

* flush when using write

* update write logic

* Revert "update write logic"

This reverts commit 5e0e611b46.

---------

Co-authored-by: chenyu <[email protected]>
 2024-10-16 14:51:41 -04:00
nimlgenandGitHub d1094fce5e amd reports on hang (#7101) 2024-10-16 21:32:44 +03:00
nimlgenandGitHub 39ab67e9ef beam capture and replay in fuzz (#7099) 
* beam capture and reply in fuzz

* clean a bit
 2024-10-16 20:26:58 +03:00
George HotzandGitHub eac58eaaba no SIGALRM on windows [pr] (#7104) 2024-10-17 00:21:04 +08:00
qazalandGitHub 9d5342976e skip TestSetitem.test_setitem_inplace_operator (#7103) 2024-10-16 18:59:08 +03:00
nimlgenandGitHub 83e7dbd89e nv fix reallocation local memory when oom (#7098) 2024-10-16 18:17:50 +03:00
George HotzandGitHub cd61e81f55 beautiful mnist works on windows (#7100) 
* beautiful mnist works on windows [pr]

* add comment for that (no pr)
 2024-10-16 23:00:05 +08:00
qazalandGitHub fc306ba89d simple to_uop [pr] (#7096) 
* simple to_uop [pr]

* fix metadata
 2024-10-16 15:52:11 +03:00
qazalandGitHub 568a4b5d92 viz touchups [pr] (#7095) 
* viz touchups [pr]

* check if port busy

* url
 2024-10-16 15:14:47 +03:00
qazalandGitHub 6172b42140 all views are UOps.VIEW [pr] (#7090) 
* all views are UOps.VIEW

* is it you

* don't recreate st uop [pr]

* first rewrite all elementwise
 2024-10-16 13:46:33 +03:00
qazalandGitHub 6acda43a2c test a rewrite of permuted reduce [pr] (#7093) 
* test a rewrite of permuted reduce [pr]

* addd rewrite tracker

* expected

* passes
 2024-10-16 12:49:54 +03:00
qazalandGitHub 56fbd408a1 viz print the sink tree as it's rewritten [pr] (#7094) 2024-10-16 11:38:27 +03:00
qazalandGitHub aadc95c2b0 don't recreate st uop [pr] (#7092) 2024-10-16 08:43:19 +03:00
qazalandGitHub 9c9c241e58 track et in viz [pr] (#7088) 2024-10-16 07:53:08 +03:00
qazalandGitHub 40f33c110b big graph var_vals as rewrite context (#7007) 
* var_vals as rewrite context

* no default arg

* add st var_vals

* delete some stuff

* add the rewrite rule again

* extra

* this whole part is preschedule

* test with a second context

* redo

* i always forget tensor variable
 2024-10-16 07:31:44 +03:00
qazalandGitHub 390171d686 delete SAVE_SCHEDULE=1 [pr] (#7087) 2024-10-16 07:13:20 +03:00
George HotzandGitHub 3169cb386d remove graph [pr] (#7085) 2024-10-16 11:40:07 +08:00
qazalandGitHub 53586eac56 late assert post permuted assign [pr] (#7084) 
* late assert post permuted assign [pr]

* a lil earlier
 2024-10-16 06:26:04 +03:00
George HotzandGitHub 023b77cc6e move MultiGraphRunner logic to GraphRunner [pr] (#7083) 
* move MultiGraphRunner logic to GraphRunner [pr]

* _access_resources
 2024-10-16 11:04:30 +08:00
qazalandGitHub 207fbc4bc7 cleanup view on reduce [pr] (#7081) 2024-10-16 05:22:52 +03:00
qazalandGitHub 067b35e915 add UOp.r [pr] (#7080) 2024-10-16 05:06:02 +03:00
George HotzandGitHub 26df50cf43 move memory_planner to memory.py [pr] (#7079) 2024-10-16 10:04:35 +08:00
qazalandGitHub bddba5897a generic elementwise view rewrite rule + merge_views (#7078) 
* generic elementwise view rewrite rule + merge_views [pr]

* no pr, views merge
 2024-10-16 04:36:21 +03:00
qazalandGitHub fb29de6cc3 split schedule to view_left and view_right [pr] (#7077) 
* split schedule to view_left and view_right [pr]

* move valid
 2024-10-16 03:39:38 +03:00