From 0f16709c00313c320a0a751ecd8101a3549ea8be Mon Sep 17 00:00:00 2001 From: George Hotz Date: Thu, 11 Apr 2024 08:37:57 -0700 Subject: [PATCH] hotfix: remove test speed vs torch --- .github/workflows/benchmark.yml | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/.github/workflows/benchmark.yml b/.github/workflows/benchmark.yml index f9ede11899..7f163a5076 100644 --- a/.github/workflows/benchmark.yml +++ b/.github/workflows/benchmark.yml @@ -187,10 +187,11 @@ jobs: run: /opt/rocm/bin/rocm-bandwidth-test - name: Run model inference benchmark run: LD_PRELOAD="/opt/rocm/lib/libhsa-runtime64.so" HSA=1 NOCLANG=1 python3 test/external/external_model_benchmark.py - - name: Test speed vs torch - run: | - python3 -c "import torch; print(torch.__version__)" - LD_PRELOAD="/opt/rocm/lib/libhsa-runtime64.so" HSA=1 BIG=2 TORCHCUDA=1 python3 test/test_speed_v_torch.py | tee torch_speed.txt + # TODO: unstable on AMD + #- name: Test speed vs torch + # run: | + # python3 -c "import torch; print(torch.__version__)" + # LD_PRELOAD="/opt/rocm/lib/libhsa-runtime64.so" HSA=1 BIG=2 TORCHCUDA=1 python3 test/test_speed_v_torch.py | tee torch_speed.txt - name: Run Tensor Core GEMM run: HSA=1 HALF=1 DEBUG=2 python3 extra/gemm/simple_matmul.py | tee matmul.txt - name: Run Stable Diffusion