George Hotz 643cbdfd50 make embedding and GPT-2 fast (#1631) ...

* make embedding fast

* jit more, variable shape support

* print mem bw

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

..

dist

distributed collectives (#1519)

2023-08-11 10:22:07 -07:00

external_copy_benchmark.py

Symbolic Shape JIT main PR (#1353)

2023-08-18 14:39:55 -07:00

external_hlb_cifar.py

Fix naming conflict with huggingface datasets (#1161)

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

external_llama_eval.py

Add option in llama.py to quantize weights to int8 at runtime (#1289)

2023-07-24 17:22:38 -07:00

external_metal_uaf.py

fix metal uaf (#964)

2023-06-09 21:28:06 -07:00

external_model_benchmark.py

single model (#1560)

2023-08-16 13:22:19 -07:00

external_multi_gpu.py

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

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

external_osx_profiling.py

multi cl_queue (#762)

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

external_test_allocator_on_models.py

move state to nn/state (#1619)

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

external_test_embedding.py

make embedding and GPT-2 fast (#1631)

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

external_test_gpu_ast.py

move to shapetracker.py

2023-03-11 07:50:07 -08:00

external_test_image.py

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

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

external_test_llvm.py

fix up external tests

2023-03-06 06:52:28 -08:00

external_test_onnx_backend.py

exclude a few more onnx tests

2023-06-19 08:51:29 -07:00

external_test_opt.py

move state to nn/state (#1619)

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

external_test_optim.py

Add LAMB optimizer (#821)

2023-05-28 15:09:05 -07:00

external_test_speed_llama.py

move state to nn/state (#1619)

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

external_test_yolo.py

test cleanups

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

external_test_yolov8.py

move state to nn/state (#1619)

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

fuzz_shapetracker.py

Add Fuzz Test symbolic / shapetracker to CI. (#1278)

2023-07-19 09:05:45 -07:00

fuzz_symbolic.py

symbolic stride (#1326)

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

graph_batchnorm.py

move state to nn/state (#1619)

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