tinygrad/extra
Fernando Vidal 73bd0b217b
add int64 as supported dtype from numpy (#699)
* add int64 as supported dtype from numpy

Without this, examples/transformer.py didn't run. With this change it runs successfully.

* Update helpers.py

* Update transformer.py

* Update training.py
2023-03-18 17:15:04 -07:00
..
gemm good changes from llama branch (#671) 2023-03-09 20:51:22 -08:00
triton A Triton backend for tinygrad (#470) 2023-02-01 11:53:57 -08:00
augment.py [Fix] fix typo: test_mnist -> datasets (#492) 2023-01-29 21:30:47 -08:00
gradcheck.py Remove Tensor.data (#565) 2023-02-18 16:36:12 -08:00
helpers.py more llama stuff 2023-03-10 10:48:10 -08:00
introspection.py Devicebufferless (#708) 2023-03-18 14:40:23 -07:00
kernel_search.py move to shapetracker.py 2023-03-11 07:50:07 -08:00
lib_test_ast.py fix ast test 2023-03-04 07:49:25 -08:00
onnx.py fine, openpilot can have 197 kernels 2023-02-27 11:48:36 -08:00
onnx_ops.py Eq magic function support (#683) 2023-03-11 10:31:46 -08:00
thneed.py dtypes nice and clean (#673) 2023-03-10 16:56:07 -08:00
training.py add int64 as supported dtype from numpy (#699) 2023-03-18 17:15:04 -07:00
utils.py Devicebufferless (#708) 2023-03-18 14:40:23 -07:00