tinygrad

sirhcm/tinygrad

Fork 0

forked from tinygrad/tinygrad

Files

T

History

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

codegen

early rewrite UOps.CONTIGUOUS (#7132 )

2024-10-17 18:35:19 +03:00

engine

Add Opt(op=OptOps.LOCAL, axis=6, amt=2) to actions (#7147 )

2024-10-18 09:03:24 -04:00

add types in all nn/init.py classes (#7002 )

2024-10-12 14:42:14 +08:00

renderer

Refactor AMD pm rules to remove handwritten bf16 bool alus (#7136 )

2024-10-18 09:00:46 +08:00