From fa2329721b68b35856d4f4adf05152afad0297e4 Mon Sep 17 00:00:00 2001 From: Anatoly Myachev Date: Fri, 29 Nov 2024 23:51:12 +0100 Subject: [PATCH] Define 'pytest-forked' and 'pytest-xdist' as 'tests' target deps Signed-off-by: Anatoly Myachev --- .github/workflows/integration-tests.yml | 6 +++--- .github/workflows/integration-tests.yml.in | 6 +++--- python/setup.py | 2 ++ 3 files changed, 8 insertions(+), 6 deletions(-) diff --git a/.github/workflows/integration-tests.yml b/.github/workflows/integration-tests.yml index 9151560ce6b9..ca7149656149 100644 --- a/.github/workflows/integration-tests.yml +++ b/.github/workflows/integration-tests.yml @@ -236,7 +236,7 @@ jobs: - name: Install pip dependencies run: | python3 -m pip install --upgrade pip - python3 -m pip install cython setuptools wheel cmake==3.24 ninja pytest-forked pytest-xdist lit + python3 -m pip install cython setuptools wheel cmake==3.24 ninja lit - name: Install Triton env: CUDA_HOME: "/usr/local/cuda" @@ -569,7 +569,7 @@ jobs: python3 -m venv ~/.venv source ~/.venv/bin/activate python3 -m pip install --upgrade pip - python3 -m pip install cython setuptools wheel cmake==3.24 ninja pytest-xdist lit pybind11 + python3 -m pip install cython setuptools wheel cmake==3.24 ninja lit pybind11 - name: Install Triton env: TRITON_BUILD_WITH_O1: "true" @@ -581,7 +581,7 @@ jobs: echo "PATH is '$PATH'" cd python ccache --zero-stats - python3 -m pip install -v --no-build-isolation . + python3 -m pip install -v --no-build-isolation .[tests] - name: CCache Stats run: ccache --print-stats - name: Inspect cache directories diff --git a/.github/workflows/integration-tests.yml.in b/.github/workflows/integration-tests.yml.in index 6d72b65207dc..64ef9b9d964c 100644 --- a/.github/workflows/integration-tests.yml.in +++ b/.github/workflows/integration-tests.yml.in @@ -268,7 +268,7 @@ jobs: - name: Install pip dependencies run: | python3 -m pip install --upgrade pip - python3 -m pip install cython setuptools wheel cmake==3.24 ninja pytest-forked pytest-xdist lit + python3 -m pip install cython setuptools wheel cmake==3.24 ninja lit - name: Install Triton env: @@ -481,7 +481,7 @@ jobs: python3 -m venv ~/.venv source ~/.venv/bin/activate python3 -m pip install --upgrade pip - python3 -m pip install cython setuptools wheel cmake==3.24 ninja pytest-xdist lit pybind11 + python3 -m pip install cython setuptools wheel cmake==3.24 ninja lit pybind11 - name: Install Triton env: TRITON_BUILD_WITH_O1: "true" @@ -493,7 +493,7 @@ jobs: echo "PATH is '$PATH'" cd python ccache --zero-stats - python3 -m pip install -v --no-build-isolation . + python3 -m pip install -v --no-build-isolation .[tests] - *print-ccache-stats - *inspect-cache-directories-step diff --git a/python/setup.py b/python/setup.py index 3d8edf507e65..a640245e7e81 100644 --- a/python/setup.py +++ b/python/setup.py @@ -752,6 +752,8 @@ def get_git_commit_hash(length=8): "isort", "numpy", "pytest", + "pytest-forked", + "pytest-xdist", "scipy>=1.7.1", "llnl-hatchet", ],