diff --git a/.github/workflows/nv-torch-latest-cpu.yml b/.github/workflows/nv-torch-latest-cpu.yml index a4832859af8c..56d2e06bca5f 100644 --- a/.github/workflows/nv-torch-latest-cpu.yml +++ b/.github/workflows/nv-torch-latest-cpu.yml @@ -14,7 +14,7 @@ concurrency: jobs: unit-tests: - runs-on: [self-hosted, cpu] + runs-on: ubuntu-20.04 steps: - uses: actions/checkout@v3 diff --git a/tests/conftest.py b/tests/conftest.py index 7e5c6fee9954..e5a8cce45fd9 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -39,14 +39,6 @@ def validate_version(expected, found): return found == expected -# This fixture prevents hangs when 2+ pytest processes try to compile the same -# code at once and causes deadlocks / hangs -@pytest.fixture(scope="session", autouse=True) -def set_torch_ext_dir(worker_id): - torch_ext_dir = os.environ["TORCH_EXTENSIONS_DIR"] - os.environ["TORCH_EXTENSIONS_DIR"] = os.path.join(torch_ext_dir, worker_id) - - @pytest.fixture(scope="session", autouse=True) def check_environment(pytestconfig): expected_torch_version = pytestconfig.getoption("torch_ver")