From 992123c428482c42af31af60d8470c0c17acf05b Mon Sep 17 00:00:00 2001 From: Logan Adams Date: Thu, 14 Mar 2024 10:28:04 -0700 Subject: [PATCH] Only run some tests --- .github/workflows/hpu-gaudi2.yml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/hpu-gaudi2.yml b/.github/workflows/hpu-gaudi2.yml index 1aaa49016000..042598b0841c 100644 --- a/.github/workflows/hpu-gaudi2.yml +++ b/.github/workflows/hpu-gaudi2.yml @@ -57,6 +57,6 @@ jobs: - name: Unit tests run: | unset TORCH_CUDA_ARCH_LIST # only jit compile for current arch + set PT_HPU_LAZY_MODE=0 cd tests - pytest -n 4 --verbose unit/ - pytest -m 'sequential' unit/ + pytest -n 4 --verbose unit/ -k "test_accelerator and test_autotuning"