Swap from conda to uv on test workflow (#368)

Author: oulgen

.github/workflows/test.yml

@@ -33,12 +33,15 @@ jobs:
       - name: Check out code
         uses: actions/checkout@v4
 
-      - name: Set up Miniconda
-        uses: conda-incubator/setup-miniconda@v3
+      - name: Install uv
+        uses: astral-sh/setup-uv@v6
         with:
-          activate-environment: torch27
           python-version: ${{ matrix.python-version }}
-          channels: defaults,pytorch,conda-forge
+          enable-cache: true
+
+      - name: Create virtual environment
+        run: |
+          uv venv --python ${{ matrix.python-version }}
 
       - name: Get current month
         id: date
@@ -48,37 +51,44 @@ jobs:
         uses: actions/cache@v4
         with:
           path: |
-            ~/.conda/pkgs
-            ~/.cache/pip
+            ~/.cache/uv
+            ~/.venv
           key: ${{ runner.os }}-deps-${{ matrix.python-version }}-${{ hashFiles('.github/workflows/test.yml', 'requirements.txt') }}-${{ steps.date.outputs.month }}
           restore-keys: |
             ${{ runner.os }}-deps-
 
       - name: Install PyTorch
         run: |
-          pip3 install -U --pre torch --index-url https://download.pytorch.org/whl/nightly/cu126
+          source .venv/bin/activate
+          uv pip install -U --pre torch --index-url https://download.pytorch.org/whl/nightly/cu126
 
       - name: Install Triton
         if: steps.cache.outputs.cache-hit != 'true'
         run: |
+          set -x
+          source .venv/bin/activate
           apt-get update
           apt-get install -y git
+          apt-get install -y gcc-13 g++-13 zlib1g-dev
+          export CC=gcc-13
+          export CXX=g++-13
           mkdir -p /tmp/$USER
           cd /tmp/$USER
-          pip uninstall -y triton pytorch-triton || true
+          uv pip uninstall triton pytorch-triton || true
           rm -rf triton/ || true
           git clone https://github.com/triton-lang/triton.git
           cd triton/
-          conda install -y -c conda-forge gcc_linux-64=13 gxx_linux-64=13 gcc=13 gxx=13
-          pip install -r python/requirements.txt
-          MAX_JOBS=$(nproc) TRITON_PARALLEL_LINK_JOBS=2 pip install .
+          uv pip install -r python/requirements.txt
+          MAX_JOBS=$(nproc) TRITON_PARALLEL_LINK_JOBS=2 uv pip install .
           cd /tmp/$USER
           rm -rf triton/
 
       - name: Install Requirements
         run: |
-          pip install -r requirements.txt
+          source .venv/bin/activate
+          uv pip install -r requirements.txt
 
       - name: Run Tests
         run: |
-          python -m unittest discover -s test/ -p "*.py" -v -t .
+          source .venv/bin/activate
+          pytest