Yixiang Gao 094d3d71be with Tensor.train() (#1935) ...

* add with.train

* remove the rest TODOs

* fix pyflake

* fix pyflake error

* fix mypy

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

..

dist

fix test_collectives (#1934)

2023-09-28 11:02:22 -07:00

external_copy_benchmark.py

add cache collector (#1595)

2023-08-28 19:59: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

move device to ops (#1646)

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

external_metal_uaf.py

fix metal uaf (#964)

2023-06-09 21:28:06 -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

multi cl_queue (#762)

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

external_test_allocator_on_models.py

no import when Python is shutting down (#1875)

2023-09-20 12:47:02 -04: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

fixes (#1893)

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

external_test_jit_on_models.py

add cache collector (#1595)

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

external_test_llvm.py

fix up external tests

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

external_test_onnx_backend.py

onnx full passing (#1076)

2023-09-05 13:23:32 -07:00

external_test_opt.py

with Tensor.train() (#1935)

2023-09-28 18:02:31 -07: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

codellama (#1702)

2023-09-02 08:45:12 -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_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

with Tensor.train() (#1935)

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