Files
tinygrad/extra/models
chenyu ad4472e6e8 cleanup llama apply_rotary_emb and other helpers (#2950)
* cleanup llama apply_rotary_emb and other helpers

used ellipsis and other higher level tensor function.
disabled the half @ half -> half tensor core as it fails uop dtype checks

* keep hip 8x8->8 wmma
2023-12-29 11:39:15 -05:00
..
2023-11-28 17:36:55 -08:00
2023-11-28 17:36:55 -08:00
2023-11-28 17:36:55 -08:00
2023-11-28 17:36:55 -08:00
2023-11-28 17:36:55 -08:00
2023-11-28 17:36:55 -08:00
2023-11-28 17:36:55 -08:00
2023-11-28 17:36:55 -08:00