no barrier side effect (#3550)

* no barrier side effect

* finish barrier removal
This commit is contained in:
George Hotz
2024-02-29 18:10:04 -08:00
committed by GitHub
parent bd9c2ced07
commit 5a6e151844
3 changed files with 4 additions and 8 deletions

View File

@@ -294,6 +294,8 @@ jobs:
run: JIT=2 METAL=1 python -m pytest -n=auto test/ --ignore=test/external --ignore=test/models --durations=20
- name: Run ONNX
run: JIT=2 METAL=1 python -m pytest -n=auto test/external/external_test_onnx_backend.py --durations=20
- name: Test tensor core ops (fake)
run: TC=2 METAL=1 DEBUG=3 python test/test_ops.py TestOps.test_gemm
- name: Test tensor core ops (real)
run: METAL=1 DEBUG=3 python test/test_ops.py TestOps.test_big_gemm
- name: Test LLaMA compile speed