From 2b6670d2eae55fbd945a7ff6b27e195a0f3cc01c Mon Sep 17 00:00:00 2001 From: chenyu Date: Thu, 4 Jan 2024 13:24:10 -0500 Subject: [PATCH] separate entry for HALF hlb_cifar10 in benchmark (#3010) --- .github/workflows/benchmark.yml | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) diff --git a/.github/workflows/benchmark.yml b/.github/workflows/benchmark.yml index dbba61d0e8..bca1314afa 100644 --- a/.github/workflows/benchmark.yml +++ b/.github/workflows/benchmark.yml @@ -135,11 +135,13 @@ jobs: HIP=1 JIT=1 python3 examples/gpt2.py --prompt "Hello." --count 10 --temperature 0 --timing | tee gpt2_jitted.txt - name: Run 10 CIFAR training steps run: STEPS=10 python3 examples/hlb_cifar10.py | tee train_cifar.txt - # TODO: make wino faster so we can enable both - - name: Run 10 CIFAR training steps w winograd - run: WINO=0 STEPS=10 python3 examples/hlb_cifar10.py | tee train_cifar_wino.txt - - name: Run 10 CIFAR training steps w WINO/HALF/HIP - run: HALF=1 HIP=1 WINO=0 STEPS=10 python3 examples/hlb_cifar10.py | tee train_cifar_wino_half_hip.txt + - name: Run 10 CIFAR training steps w HALF + run: HALF=1 STEPS=10 python3 examples/hlb_cifar10.py | tee train_cifar_half.txt + # # TODO: make wino faster so we can enable both + # - name: Run 10 CIFAR training steps w winograd + # run: WINO=1 STEPS=10 python3 examples/hlb_cifar10.py | tee train_cifar_wino.txt + # - name: Run 10 CIFAR training steps w WINO/HALF/HIP + # run: HALF=1 WINO=1 STEPS=10 python3 examples/hlb_cifar10.py | tee train_cifar_wino_half_hip.txt - uses: actions/upload-artifact@v3 with: name: Speed (AMD) @@ -147,6 +149,7 @@ jobs: onnx_inference_speed.csv torch_speed.txt train_cifar.txt + train_cifar_half.txt train_cifar_wino.txt train_cifar_wino_half_hip.txt llama_unjitted.txt