nimlgen 1c0449e190 add cache collector (#1595) ...

* init cache collector

* add test_cache_collector.py

* switch GlobalCounters.cache to CacheCollector

* init jit models test

* jitted SD

* add debug msg to print loaded bufs count

* moved cache collctor to jit

* clearer SD

* no double device import

2023-08-28 19:59:55 -07:00

..

accel

Revert "update editorconfig, enforce via CI (#1343)" (#1380)

2023-07-31 10:35:50 -07:00

assembly

PTX Reintegration and Passing Tests (#1512)

2023-08-16 16:20:20 -07:00

datasets

CIFAR 94.03% (#1340)

2023-08-08 15:13:24 -07:00

disk

Fast DiskTensor to other Tensor (#916)

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

dist

add cache collector (#1595)

2023-08-28 19:59:55 -07:00

gemm

move device to ops (#1646)

2023-08-23 08:30:17 -07:00

intel

Revert "update editorconfig, enforce via CI (#1343)" (#1380)

2023-07-31 10:35:50 -07:00

ptx

ptx example

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

rocm

fast HIP gemm -> 100 TFLOPS (#1476)

2023-08-09 06:54:15 -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 naming conflict with huggingface datasets (#1161)

2023-07-07 10:43:44 -07:00

export_model.py

move state to nn/state (#1619)

2023-08-22 07:36:24 -07:00

gradcheck.py

Fix: Jacobian tests [WIP] (#1126)

2023-07-05 15:36:22 -07:00

helpers.py

helpers.py: improved test coverage + exception handling (#1165)

2023-07-07 10:26:05 -07:00

hip_wrapper.py

fix: small fix for lru on hip (#1567)

2023-08-18 09:18:38 -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

Add pylint trailing whitespace rule (#1314)

2023-07-21 13:37:55 -04:00

onnx_ops.py

Run onnx commavq model (#1537)

2023-08-16 12:24:40 -07:00

onnx.py

Run onnx commavq model (#1537)

2023-08-16 12:24:40 -07:00

thneed.py

add cache collector (#1595)

2023-08-28 19:59:55 -07:00

training.py

np.argmax to Tensor.argmax (#1608)

2023-08-21 15:22:29 -07:00

utils.py

move device to ops (#1646)

2023-08-23 08:30:17 -07:00