Skip to content

Commit

Permalink
Revert "Prevent hangs in CI during parallel run compilation (#2844)" (#…
Browse files Browse the repository at this point in the history
…3817)

This reverts commit 2b2be85.
  • Loading branch information
jeffra authored Jun 26, 2023
1 parent 2b2be85 commit 6102d12
Show file tree
Hide file tree
Showing 2 changed files with 1 addition and 9 deletions.
2 changes: 1 addition & 1 deletion .github/workflows/nv-torch-latest-cpu.yml
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@ concurrency:

jobs:
unit-tests:
runs-on: [self-hosted, cpu]
runs-on: ubuntu-20.04

steps:
- uses: actions/checkout@v3
Expand Down
8 changes: 0 additions & 8 deletions tests/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -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")
Expand Down

0 comments on commit 6102d12

Please sign in to comment.