github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-29 03:00:14 -04:00
Code Issues Packages Projects Releases Wiki Activity

Reenable tensor cores for self-hosted Mac CI (#1717)

Browse Source 
* debug 5 matmul

* allow tensor cores in CI

* tensor cores on arm64

* put debug back
This commit is contained in:
George Hotz
2023-08-30 07:53:04 -07:00
committed by GitHub
parent ac183568be
commit fdd7f282cb
3 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
.github/workflows/benchmark.yml vendored
View File

@@ -22,7 +22,9 @@ jobs:
run: BIG=2 MPS=1 python3 test/test_speed_v_torch.py | tee torch_speed.txt
shell: bash
- name: Run Tensor Core GEMM
run: DEBUG=2 python3 extra/gemm/simple_matmul.py | tee matmul.txt
run: |
ln -s ~/tinygrad/disassemblers/applegpu disassemblers/applegpu
DEBUG=2 python3 extra/gemm/simple_matmul.py | tee matmul.txt
shell: bash
- name: Run Stable Diffusion
run: |