github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-22 21:38:10 -05:00
Code Issues Packages Projects Releases Wiki Activity
Files
late_codegen
tinygrad/extra/models/transformer.py
chenyu 1692087db5 _one_hot_along_dim input needs to be int (#9179) 
* _one_hot_along_dim input needs to be int

indexing and onehot compare with arange, and non-int dtype is likely a bug
2025-02-20 09:00:43 -05:00

2.7 KiB
Raw Permalink Blame History

View Raw
Reference in New Issue View Git Blame Copy Permalink