github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-10 07:28:15 -05:00
Code Issues Packages Projects Releases Wiki Activity

HIP=1 NOCLANG=1 for tinybox external_model_benchmark (#3270)

Browse Source 
used HIP instead of GPU and disabled slow CLANG
This commit is contained in:
chenyu
2024-01-28 22:05:26 -05:00
committed by GitHub
parent 085dc87bed
commit 34c7621556
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -122,7 +122,7 @@ jobs:
ln -s ~/tinygrad/weights/LLaMA weights/LLaMA
ln -s ~/tinygrad/extra/datasets/cifar-10-python.tar.gz extra/datasets/cifar-10-python.tar.gz
- name: Run model inference benchmark
run: GPU=1 python3 test/external/external_model_benchmark.py
run: HIP=1 NOCLANG=1 python3 test/external/external_model_benchmark.py
- name: Test speed vs torch
run: BIG=2 TORCHCUDA=1 python3 test/test_speed_v_torch.py | tee torch_speed.txt
- name: Run Tensor Core GEMM