tinygrad

mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-24 06:18:01 -05:00

Files

qazal 12996d3a7d green linearizer asserts for ops (#2800 )

* these asserts should pass

* fix that assert

* ALU dtypes

* acc dtype for group_for_reduce

* cast image ALUs to the base dtype

* remove all casts from linearizer

* fix argmax

* fix multinomial

* fix __getitem__

* Revert "fix __getitem__"

This reverts commit 62ad719bfa.

* fix MemBuffer outputs being wrong when there is an arange + ALU with a different dtype

eg. fancy slicing (int, float), bert embeddings (int, long)

this should be fixed in lazy instead of having to break the kernel

* cleanup argmax fix

* fix matmul in ints

cast in the end

* fix llama

* skip wrong hardcoded asts in the worlds dataset

* fix llama p2

* cleanup missing parts of the diff

---------

Co-authored-by: George Hotz <geohot@gmail.com>

2023-12-25 10:41:54 -05:00

extract_policynet.py

merge kernel and optimizer (#2200 )

2023-11-01 15:20:01 -07:00

extract_sa_pairs.py

merge kernel and optimizer (#2200 )

2023-11-01 15:20:01 -07:00

generate_dataset.sh

op logger + replay (#2021 )

2023-10-08 15:10:18 -07:00

get_action_space.py

merge kernel and optimizer (#2200 )