@@ -139,7 +139,8 @@ jobs:
139139 run : |
140140 torch_version=$(python -c "import torch; print(torch.__version__)")
141141 torch_git_version=$(python -c "import torch; print(torch.version.git_version)")
142- torchbench_version=$(curl https://raw.githubusercontent.com/pytorch/pytorch/${torch_git_version}/.github/ci_commit_pins/torchbench.txt)
142+ # torchbench_version=$(curl https://raw.githubusercontent.com/pytorch/pytorch/${torch_git_version}/.github/ci_commit_pins/torchbench.txt)
143+ torchbench_version=main
143144 torchvision_version=$(curl https://raw.githubusercontent.com/pytorch/pytorch/${torch_git_version}/.github/ci_commit_pins/vision.txt)
144145 torchaudio_version=$(curl https://raw.githubusercontent.com/pytorch/pytorch/${torch_git_version}/.github/ci_commit_pins/audio.txt)
145146 hf_version=$(curl https://raw.githubusercontent.com/pytorch/pytorch/${torch_git_version}/.ci/docker/ci_commit_pins/huggingface.txt)
@@ -210,10 +211,10 @@ jobs:
210211 - name : Run benchmarks
211212 working-directory : benchmark
212213 run : |
214+ export PT_HPU_LAZY_MODE=0
213215 echo "Run torchbench"
214- touch gaudi_hpu_benchmark.json
215- # python run_benchmark.py test_bench --accuracy --device hpu --test eval \
216- # --output gaudi_hpu_benchmark.json
216+ python run_benchmark.py test_bench --accuracy --device hpu --test eval \
217+ --output gaudi_hpu_benchmark.json
217218
218219 - name : Upload the benchmark report file
219220 id : upload-report
@@ -228,7 +229,7 @@ jobs:
228229 - name : Write to workflow job summary
229230 run : |
230231 echo "Write to workflow job summary"
231- # python .ci/benchmark.py --write-gh-job-summary --path benchmark/gaudi_hpu_benchmark.json
232+ python .ci/benchmark.py --write-gh-job-summary --path benchmark/gaudi_hpu_benchmark.json
232233
233234 - name : Write to workflow job summary
234235 run : |
0 commit comments