George Hotz bd8a5c2ced Simple CUDA Runtime (#480) ...

* factor out opencl runtime

* don't use CL outside the runtime

* cuda runtime adds

* final_dimension

* tests pass with CUDA backend

* more cuda

* cuda simpler

* retain old functionality

* linter and typing

* move globalcounters out of runtimes

* oops, GlobalCounters in cuda

* MAX_OUTPUT_SHAPE=3 is fine for CUDA

2023-01-27 16:26:24 -08:00

..

efficientnet

add ViT test and car

2022-06-05 17:12:43 -07:00

__init__.py

All devices are equal! (#196)

2020-12-15 23:44:08 -08:00

Dockerfile

testdocker

2023-01-09 12:41:52 -08:00

external_test_gpu_ast.py

first little attempt at search

2023-01-25 11:49:29 -08:00

external_test_image.py

simplify ones after axis splitting

2023-01-14 10:51:43 -08:00

external_test_llvm.py

LLVM Backend take 2 (#403)

2022-10-26 20:32:31 -07:00

external_test_opt.py

one more opt test

2022-10-28 18:37:53 -07:00

graph_batchnorm.py

Simple version of the new GPU backend (#458)

2023-01-10 19:16:02 -08:00

lib_test_ast.py

DeviceBuffer has abstract methods

2023-01-25 19:16:23 -08:00

test_conv.py

Fold reduce (#362)

2022-07-19 09:24:02 -07:00

test_efficientnet.py

test_onnx works with enet also

2022-06-11 14:30:26 -07:00

test_gc.py

set requires_grad to None (#387)

2022-09-21 11:16:02 -04:00

test_mnist.py

Fix weight init: this work? (#391)

2022-09-25 16:46:33 -04:00

test_net_speed.py

import tests from CL metal texture fix

2022-09-19 20:01:47 -04:00

test_nn.py

Revert "Rename Normalize and move to nn (#415)" (#474)

2023-01-25 07:50:04 -08:00

test_onnx.py

support float16 onnx weights (#384)

2022-09-15 09:12:18 -04:00

test_ops.py

Simple CUDA Runtime (#480)

2023-01-27 16:26:24 -08:00

test_optim.py

tinygrad.nn (#367)

2022-08-18 07:41:00 -07:00

test_shapetracker.py

move shapetracker (#466)

2023-01-19 09:56:31 -08:00

test_speed_v_torch.py

touch up pytorch speed tests

2023-01-25 18:11:26 -08:00

test_symbolic.py

Typing (#468)

2023-01-21 09:09:22 -08:00

test_tensor.py

set requires_grad to None (#387)

2022-09-21 11:16:02 -04:00

test_train.py

Simple version of the new GPU backend (#458)

2023-01-10 19:16:02 -08:00