George Hotz 3baaf298d6 two stage cumsum in tensor.py (#2331) ...

* two stage cumsum in tensor.py

* 2 more kernels for llama cumsum

* gpt-2 and llama use fast multinomial

2023-11-16 12:09:53 -08:00

..

dist

metal indirect command buffers (#2285)

2023-11-13 17:58:26 -08:00

external_copy_benchmark.py

fix some tests related to JitItem (#2279)

2023-11-11 23:00:35 -05:00

external_llama_eval.py

move device to ops (#1646)

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

external_model_benchmark.py

fix try except not catching fxn() in benchmark (#1783)

2023-09-06 07:36:43 -07:00

external_multi_gpu.py

revert typo in external_multi_gpu.py (#1777)

2023-09-05 20:46:28 -07:00

external_osx_profiling.py

adapting test/external/external_osx_profiling.py to the new code base (#2002)

2023-10-08 05:55:00 -07:00

external_test_allocator_on_models.py

move optimize_local_size (#2221)

2023-11-05 21:00:52 -08:00

external_test_embedding.py

make embedding and GPT-2 fast (#1631)

2023-08-22 15:14:38 -07:00

external_test_image.py

fixes (#1893)

2023-09-22 07:20:27 +08:00

external_test_jit_on_models.py

JIT support in Interpreted (#2314)

2023-11-15 11:13:38 -08:00

external_test_onnx_backend.py

Tensor(None) is Tensor([]) (#2316)

2023-11-15 13:49:18 -05:00

external_test_opt.py

two stage cumsum in tensor.py (#2331)

2023-11-16 12:09:53 -08:00

external_test_optim.py

use class Foo: instead of class Foo(): (#1797)

2023-09-06 12:20:25 -07:00

external_test_speed_llama.py

llama take int and convert to Variable internally (#2284)

2023-11-12 17:11:37 -05:00

external_test_uops_graphing.py

uops graphing + phi (#2120)

2023-10-19 22:26:28 -07:00

external_test_yolo.py

test cleanups

2023-03-03 06:36:06 -08:00

external_test_yolov8.py

[ready] Replacing os with pathlib (#1708)

2023-08-30 10:41:08 -07:00

fuzz_linearizer.py

fuzz_linearizer same api for interpreted and compiled (#2320)

2023-11-15 17:40:22 -05:00

fuzz_shapetracker.py

remove real_offset (#2234)

2023-11-07 17:30:53 -08:00

fuzz_symbolic.py

symbolic stride (#1326)

2023-07-23 12:41:22 -07:00

graph_batchnorm.py

with Tensor.train() (#1935)

2023-09-28 18:02:31 -07:00

test_example.py

external (#2191)

2023-10-31 13:57:24 -07:00