minor fixes

csukuangfj · Feb 1, 2025 · 798a1f3 · 798a1f3
1 parent 2569bf6
commit 798a1f3
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 6 deletions.
diff --git a/.github/workflows/ubuntu-cuda-wheels.yml b/.github/workflows/ubuntu-cuda-wheels.yml
@@ -26,11 +26,11 @@ jobs:
         id: set-matrix
         run: |
           # outputting for debugging purposes
-          # python ./scripts/github_actions/generate_build_matrix.py --enable-cuda
-          # MATRIX=$(python ./scripts/github_actions/generate_build_matrix.py --enable-cuda)
+          python ./scripts/github_actions/generate_build_matrix.py --enable-cuda
+          MATRIX=$(python ./scripts/github_actions/generate_build_matrix.py --enable-cuda)
 
-          python ./scripts/github_actions/generate_build_matrix.py --enable-cuda --test-only-latest-torch
-          MATRIX=$(python ./scripts/github_actions/generate_build_matrix.py --enable-cuda --test-only-latest-torch)
+          # python ./scripts/github_actions/generate_build_matrix.py --enable-cuda --test-only-latest-torch
+          # MATRIX=$(python ./scripts/github_actions/generate_build_matrix.py --enable-cuda --test-only-latest-torch)
           echo "::set-output name=matrix::${MATRIX}"
 
   build-manylinux-wheels:
@@ -102,10 +102,13 @@ jobs:
               # cp313-cp313t  (no gil)
               if [[ $PYTHON_VERSION == "3.6" ]]; then
                 python_dir=/opt/python/cp36-cp36m
+                export PYTHONPATH=/opt/python/cp36-cp36m/lib/python3.6/site-packages:$PYTHONPATH
               elif [[ $PYTHON_VERSION == "3.7" ]]; then
                 python_dir=/opt/python/cp37-cp37m
+                export PYTHONPATH=/opt/python/cp37-cp37m/lib/python3.7/site-packages:$PYTHONPATH
               elif [[ $PYTHON_VERSION == "3.8" ]]; then
                 python_dir=/opt/python/cp38-cp38
+                export PYTHONPATH=/opt/python/cp38-cp38/lib/python3.8/site-packages:$PYTHONPATH
               elif [[ $PYTHON_VERSION == "3.9" ]]; then
                 python_dir=/opt/python/cp39-cp39
                 export PYTHONPATH=/opt/python/cp39-cp39/lib/python3.9/site-packages:$PYTHONPATH

diff --git a/scripts/github_actions/generate_build_matrix.py b/scripts/github_actions/generate_build_matrix.py
@@ -288,8 +288,7 @@ def generate_build_matrix(enable_cuda, for_windows, for_macos, test_only_latest_
         "2.6.0": {
             "python-version": ["3.9", "3.10", "3.11", "3.12", "3.13"],
             "cuda": (
-                #  ["11.8", "12.4", "12.6"]  # default 12.4
-                ["12.6"]  # default 12.4
+                ["11.8", "12.4", "12.6"]  # default 12.4
                 if not for_windows
                 else ["11.8.0", "12.4.0", "12.6.0"]
             ),