From 34c76215564c403902cb05094778dfdade7e5ae6 Mon Sep 17 00:00:00 2001 From: chenyu Date: Sun, 28 Jan 2024 22:05:26 -0500 Subject: [PATCH] HIP=1 NOCLANG=1 for tinybox external_model_benchmark (#3270) used HIP instead of GPU and disabled slow CLANG --- .github/workflows/benchmark.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/benchmark.yml b/.github/workflows/benchmark.yml index a8a8c0d3c7..5a91e79d0f 100644 --- a/.github/workflows/benchmark.yml +++ b/.github/workflows/benchmark.yml @@ -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