diff --git a/.github/workflows/benchmark.yml b/.github/workflows/benchmark.yml index b8a2c84b09..a1fa66616a 100644 --- a/.github/workflows/benchmark.yml +++ b/.github/workflows/benchmark.yml @@ -58,8 +58,8 @@ jobs: run: STEPS=10 python3 examples/hlb_cifar10.py | tee train_cifar.txt - name: Run 10 CIFAR training steps w HALF run: STEPS=10 DEFAULT_FLOAT=HALF python3 examples/hlb_cifar10.py | tee train_cifar_half.txt - - name: Run 10 CIFAR training steps w BF16 - run: JIT=2 STEPS=10 DEFAULT_FLOAT=BFLOAT16 python3 examples/hlb_cifar10.py | tee train_cifar_bf16.txt + #- name: Run 10 CIFAR training steps w BF16 + # run: STEPS=10 DEFAULT_FLOAT=BFLOAT16 python3 examples/hlb_cifar10.py | tee train_cifar_bf16.txt # TODO: this is flaky too # - name: Run 10 CIFAR training steps w winograd # run: WINO=1 STEPS=10 python3 examples/hlb_cifar10.py | tee train_cifar_wino.txt