From 798a1f32d344f7dae10171645775b47e03abdc47 Mon Sep 17 00:00:00 2001
From: Fangjun Kuang <csukuangfj@gmail.com>
Date: Sat, 1 Feb 2025 21:17:06 +0800
Subject: [PATCH] minor fixes

---
 .github/workflows/ubuntu-cuda-wheels.yml        | 11 +++++++----
 scripts/github_actions/generate_build_matrix.py |  3 +--
 2 files changed, 8 insertions(+), 6 deletions(-)

diff --git a/.github/workflows/ubuntu-cuda-wheels.yml b/.github/workflows/ubuntu-cuda-wheels.yml
index 219968e..bd20b65 100644
--- 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
index f9cd94a..b76a930 100755
--- 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"]
             ),