fix CI

Author: FindDefinition

.github/workflows/build.yaml

@@ -115,12 +115,12 @@ jobs:
         if: (github.event_name == 'push' && (startsWith(github.ref, 'refs/tags')) && (env.CUDA_VERSION != '') ) || env.CUDA_VERSION == '126'
         run: |
           # clone nvidia cuda cccl to third_party/
-          if [ $CUDA_VERSION -lt "12.0" ]; then
+          if [ $CUDA_VERSION -lt "120" ]; then
               CCCL_BRANCH_NAME="2.7.0"
-          elif [ $CUDA_VERSION -lt "12.2" ]; then
+          elif [ $CUDA_VERSION -lt "122" ]; then
               CCCL_BRANCH_NAME="2.2.0"
           else
-              CCCL_MINOR_VERSION=$(echo $CUDA_VERSION | cut -d'.' -f2)
+              CCCL_MINOR_VERSION=${CUDA_VERSION:2:1}
               CCCL_BRANCH_NAME="2.$CCCL_MINOR_VERSION.0"
           fi
           git clone https://github.com/NVIDIA/cccl.git third_party/cccl -b v$CCCL_BRANCH_NAME

cumm/common.py

@@ -426,7 +426,7 @@ def __init__(self):
 class CompileInfo(pccm.Class):
     def __init__(self):
         super().__init__()
-        if not CUMM_CPU_ONLY_BUILD:
+        if not CUMM_CPU_ONLY_BUILD and not compat.IsAppleSiliconMacOs:
             include, lib64 = _get_cuda_include_lib()
             _, self.cuda_archs, self.has_ptx = _get_cuda_arch_flags()
             self.ptx_arch = self.cuda_archs[-1]

tools/build-wheels-dev.sh

@@ -13,21 +13,20 @@ function repair_wheel {
 
 export CUMM_DISABLE_JIT="1"
 export CUMM_CUDA_ARCH_LIST="7.5;8.6"
-export CUDA_VERSION="12.6"
+export CUDA_VERSION="126"
 if [ -d /io/third_party/cccl ]; then
   echo "CCCL already cloned."
 else
     # clone nvidia cuda cccl to cumm/third_party
-    if [ $CUDA_VERSION -lt "12.0" ]; then
+    if [ $CUDA_VERSION -lt "120" ]; then
         CCCL_BRANCH_NAME="2.7.0"
-    elif [ $CUDA_VERSION -lt "12.2" ]; then
+    elif [ $CUDA_VERSION -lt "122" ]; then
         CCCL_BRANCH_NAME="2.2.0"
     else
-        CCCL_MINOR_VERSION=$(echo $CUDA_VERSION | cut -d'.' -f2)
+        CCCL_MINOR_VERSION=${CUDA_VERSION:2:1}
         CCCL_BRANCH_NAME="2.$CCCL_MINOR_VERSION.0"
     fi
     git clone https://github.com/NVIDIA/cccl.git /io/third_party/cccl -b v$CCCL_BRANCH_NAME
-
 fi
 
 "/opt/python/cp311-cp311/bin/pip" wheel /io/ --no-deps -w /io/wheelhouse_tmp -v