From 0d5845fb5bd058903547b864dba00a4fe8946390 Mon Sep 17 00:00:00 2001 From: George Hotz Date: Tue, 26 Mar 2024 20:44:05 -0700 Subject: [PATCH] hotfix: jit is flaky on mac --- .github/workflows/benchmark.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/benchmark.yml b/.github/workflows/benchmark.yml index 375f5f1384..b8a2c84b09 100644 --- a/.github/workflows/benchmark.yml +++ b/.github/workflows/benchmark.yml @@ -59,7 +59,7 @@ jobs: - 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: STEPS=10 DEFAULT_FLOAT=BFLOAT16 python3 examples/hlb_cifar10.py | tee train_cifar_bf16.txt + run: JIT=2 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