diff --git a/.github/workflows/benchmark.yml b/.github/workflows/benchmark.yml index cccf231dee..dcac986702 100644 --- a/.github/workflows/benchmark.yml +++ b/.github/workflows/benchmark.yml @@ -373,6 +373,10 @@ jobs: # run: HSA=1 M_START=12 M_STOP=20 M_STEP=1 N_START=12 N_STOP=20 N_STEP=1 K_START=28 K_STOP=36 K_STEP=1 HALF=1 TC_OPT=2 DEBUG=2 python3 ./extra/gemm/fuzz_matmul.py - name: Remove amdgpu run: sleep 5 && sudo rmmod amdgpu # sleep a bit to let the driver unload the prev pid. + - name: Test AM cold start time + run: time AMD=1 AM_RESET=1 python3 test/test_tiny.py TestTiny.test_plus + - name: Test AM warm start time + run: time AMD=1 python3 test/test_tiny.py TestTiny.test_plus - name: Run Stable Diffusion run: AMD=1 python3 examples/stable_diffusion.py --fp16 --seed 0 --noshow --timing | tee sd.txt - name: Run SDXL