From 373660a1768b32f43cd8e363bbd7ccd999773471 Mon Sep 17 00:00:00 2001 From: Logan Adams Date: Mon, 16 Dec 2024 15:08:14 -0800 Subject: [PATCH] Update to repro failures on --- .github/workflows/nv-torch-latest-v100.yml | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/.github/workflows/nv-torch-latest-v100.yml b/.github/workflows/nv-torch-latest-v100.yml index a1ba4937d164..9c0c8d1ad8ce 100644 --- a/.github/workflows/nv-torch-latest-v100.yml +++ b/.github/workflows/nv-torch-latest-v100.yml @@ -55,5 +55,4 @@ jobs: run: | unset TORCH_CUDA_ARCH_LIST # only jit compile for current arch cd tests - pytest $PYTEST_OPTS --forked -n 4 unit/ --torch_ver="2.5" --cuda_ver="12.1" - pytest $PYTEST_OPTS --forked -m 'sequential' unit/ --torch_ver="2.5" --cuda_ver="12.1" + pytest $PYTEST_OPTS --forked -m 'inference_ops' unit/ --torch_ver="2.5" --cuda_ver="12.1"