Alex Wang 3d63c71e27 HIP backend (#750) ...

* llama works for HIP backend

* Use hipMemcpyAsync; Less lines of code

* Remove unused code

* Refactor

* Add comments; hipDeviceSynchronize

* HIP over GPU; Remove PyHIP dependency

* Cleanups

* Fix mypy check

* Merge master; Dump assembly code

2023-06-18 11:35:57 -07:00

..

disk

Fast DiskTensor to other Tensor (#916)

2023-06-03 12:25:41 -07:00

gemm

Add tensorflow GEMM benchmark script (#1000)

2023-06-18 10:57:45 -07:00

intel

intel benchmark matmul gets 60 TFLOPS?

2023-06-04 17:01:50 +00:00

ptx

ptx example

2023-05-26 19:28:51 -07:00

rocm

faster RDNA assembly backend (#990)

2023-06-16 12:06:38 -07:00

triton

A Triton backend for tinygrad (#470)

2023-02-01 11:53:57 -08:00

archprobe.py

no werror in archprobe

2023-05-03 19:34:17 +00: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

RDNA assembly backend ($1000 bounty) (#787)

2023-06-16 09:33:18 -07:00

hip_wrapper.py

HIP backend (#750)

2023-06-18 11:35:57 -07: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

lr_scheduler.py

LR Schedulers (#755)

2023-05-27 07:47:49 -07:00

onnx_ops.py

Adds floor/ceil (#989)

2023-06-17 10:56:21 -07:00

onnx.py

fixes to Onnx ops LayerNormalization/Prelu and added OptionalHasElement/OptionalGetElement (#956)

2023-06-08 16:09:19 -07:00

thneed.py

multi cl_queue (#762)

2023-05-03 12:15:28 -07:00

training.py

add int64 as supported dtype from numpy (#699)

2023-03-18 17:15:04 -07:00

utils.py

Refactor LoadOps (#910)

2023-06-03 09:40:43 -07:00