Rebuild for CUDA 12 w/arch + Windows support (#13)

automerged PR by conda-forge/automerge-action

.ci_support/migrations/cuda120.yaml

@@ -3,7 +3,7 @@ __migrator:
   kind:
     version
   migration_number:
-    2
+    3
   build_number:
     1
   paused: false
@@ -47,13 +47,13 @@ __migrator:
       - quay.io/condaforge/linux-anvil-cos7-x86_64        # [linux64 and os.environ.get("BUILD_PLATFORM") == "linux-64"]
     cuda_compiler_version:
       - None
-      - 10.2                       # [(linux64 or win) and os.environ.get("CF_CUDA_ENABLED", "False") == "True"]
-      - 11.0                       # [(linux64 or win) and os.environ.get("CF_CUDA_ENABLED", "False") == "True"]
-      - 11.1                       # [(linux64 or win) and os.environ.get("CF_CUDA_ENABLED", "False") == "True"]
-      - 11.2                       # [(linux or win) and os.environ.get("CF_CUDA_ENABLED", "False") == "True"]
-      - 12.0                       # [(linux or win) and os.environ.get("CF_CUDA_ENABLED", "False") == "True"]
+      - 10.2                       # [(linux64 or win64) and os.environ.get("CF_CUDA_ENABLED", "False") == "True"]
+      - 11.0                       # [(linux64 or win64) and os.environ.get("CF_CUDA_ENABLED", "False") == "True"]
+      - 11.1                       # [(linux64 or win64) and os.environ.get("CF_CUDA_ENABLED", "False") == "True"]
+      - 11.2                       # [(linux or win64) and os.environ.get("CF_CUDA_ENABLED", "False") == "True"]
+      - 12.0                       # [(linux or win64) and os.environ.get("CF_CUDA_ENABLED", "False") == "True"]
   commit_message: |
-    Rebuild for CUDA 12 w/arch support
+    Rebuild for CUDA 12 w/arch + Windows support
     
     The transition to CUDA 12 SDK includes new packages for all CUDA libraries and
     build tools. Notably, the cudatoolkit package no longer exists, and packages
@@ -62,11 +62,11 @@ __migrator:
     [see this issue]( https://github.com/conda-forge/conda-forge.github.io/issues/1963 ).
     Please feel free to raise any issues encountered there. Thank you! :pray:
 
-cuda_compiler:                 # [linux and os.environ.get("CF_CUDA_ENABLED", "False") == "True"]
-  - cuda-nvcc                  # [linux and os.environ.get("CF_CUDA_ENABLED", "False") == "True"]
+cuda_compiler:                 # [(linux or win64) and os.environ.get("CF_CUDA_ENABLED", "False") == "True"]
+  - cuda-nvcc                  # [(linux or win64) and os.environ.get("CF_CUDA_ENABLED", "False") == "True"]
 
-cuda_compiler_version:         # [linux and os.environ.get("CF_CUDA_ENABLED", "False") == "True"]
-  - 12.0                       # [linux and os.environ.get("CF_CUDA_ENABLED", "False") == "True"]
+cuda_compiler_version:         # [(linux or win64) and os.environ.get("CF_CUDA_ENABLED", "False") == "True"]
+  - 12.0                       # [(linux or win64) and os.environ.get("CF_CUDA_ENABLED", "False") == "True"]
 
 c_compiler_version:            # [linux and os.environ.get("CF_CUDA_ENABLED", "False") == "True"]
   - 12                         # [linux and os.environ.get("CF_CUDA_ENABLED", "False") == "True"]

recipe/bld.bat

@@ -1,12 +1,27 @@
 @echo on
 
 :: This step is required when building from raw source archive
-make generate --jobs %CPU_COUNT%
-if errorlevel 1 exit /b 1
+:: make generate --jobs %CPU_COUNT%
+:: if errorlevel 1 exit /b 1
 
 :: Duplicate lists because of https://bitbucket.org/icl/magma/pull-requests/32
-set "CUDA_ARCH_LIST=sm_35,sm_50,sm_60,sm_70,sm_80"
-set "CUDAARCHS=35-virtual;50-virtual;60-virtual;70-virtual;80-virtual"
+set "CUDA_ARCH_LIST=sm_50,sm_60,sm_70,sm_80"
+set "CUDAARCHS=50-virtual;60-virtual;70-virtual;80-virtual"
+
+if "%cuda_compiler_version%"=="11.2" (
+  set "CUDA_ARCH_LIST=sm_35,%CUDA_ARCH_LIST%"
+  set "CUDAARCHS=35-virtual;%CUDAARCHS%"
+)
+
+if "%cuda_compiler_version%"=="11.8" (
+  set "CUDA_ARCH_LIST=sm_35,%CUDA_ARCH_LIST%,sm_90"
+  set "CUDAARCHS=35-virtual;%CUDAARCHS%;90-virtual"
+)
+
+if "%cuda_compiler_version%"=="12.0" (
+  set "CUDA_ARCH_LIST=%CUDA_ARCH_LIST%,sm_90"
+  set "CUDAARCHS=%CUDAARCHS%;90-virtual"
+)
 
 md build
 cd build

recipe/build.sh

@@ -31,17 +31,11 @@ fi
 if [[ "$target_platform" == "linux-ppc64le" ]]; then
   export CUDA_ARCH_LIST=${CUDA_ARCH_LIST//,sm_89/}
   export CUDAARCHS=${CUDAARCHS//;89-real/}
-  if [[ "$cuda_compiler_version" == "11"* ]]; then
-    export CMAKE_ARGS="${CMAKE_ARGS} -DCUDAToolkit_ROOT=/usr/local/cuda/targets/ppc64le-linux"
-  fi
 fi
 
 if [[ "$target_platform" == "linux-aarch64" ]]; then
   export CUDA_ARCH_LIST=${CUDA_ARCH_LIST//,sm_89/}
   export CUDAARCHS=${CUDAARCHS//;89-real/}
-  if [[ "$cuda_compiler_version" == "11"* ]]; then
-    export CMAKE_ARGS="${CMAKE_ARGS} -DCUDAToolkit_ROOT=/usr/local/cuda/targets/sbsa-linux"
-  fi
 fi
 
 # Remove CXX standard flags added by conda-forge. std=c++11 is required to

recipe/meta.yaml

@@ -98,11 +98,12 @@ source:
       - optional-install.patch
 
 build:
-  number: 0
+  number: 1
   skip: true  # [cuda_compiler_version == "None"]
   skip: true  # [cuda_compiler_version == "10.2"]
   skip: true  # [cuda_compiler_version == "11.0"]
   skip: true  # [cuda_compiler_version == "11.1"]
+  skip: true  # [win and (cuda_compiler_version == "12.0")]
   run_exports:
     - {{ pin_subpackage('libmagma', max_pin='x.x.x') }}