From 7f1802cd50946a7b4c8f58c654f8fded28800821 Mon Sep 17 00:00:00 2001 From: Yixiang Gao Date: Wed, 3 Jan 2024 09:09:34 -0800 Subject: [PATCH] update benchmark --- .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 346cb0f0..44fe6827 100644 --- a/.github/workflows/benchmark.yml +++ b/.github/workflows/benchmark.yml @@ -121,10 +121,11 @@ jobs: run: HIP=1 HALF=1 DEBUG=2 python3 extra/gemm/simple_matmul.py | tee matmul.txt - name: Run Stable Diffusion run: python3 examples/stable_diffusion.py --seed 0 --noshow --timing | tee sd.txt - - name: Run LLaMA - run: | - JIT=0 python3 examples/llama.py --gen 1 --prompt "Hello." --count 10 --temperature 0 --timing | tee llama_unjitted.txt - JIT=1 python3 examples/llama.py --gen 1 --prompt "Hello." --count 10 --temperature 0 --timing | tee llama_jitted.txt + # TODO: rocm 6.0 broke this + # - name: Run LLaMA + # run: | + # JIT=0 python3 examples/llama.py --gen 1 --prompt "Hello." --count 10 --temperature 0 --timing | tee llama_unjitted.txt + # JIT=1 python3 examples/llama.py --gen 1 --prompt "Hello." --count 10 --temperature 0 --timing | tee llama_jitted.txt - name: Run GPT2 (with HIP) run: | HIP=1 JIT=0 python3 examples/gpt2.py --prompt "Hello." --count 10 --temperature 0 --timing | tee gpt2_unjitted.txt