Add Windows + CUDA (#6)

automerged PR by conda-forge/automerge-action
conda-forge · Dec 5, 2023 · 29bd539 · 29bd539
2 parents a8ab716 + 403f840
commit 29bd539
Show file tree

Hide file tree

Showing 8 changed files with 272 additions and 7 deletions.
diff --git a/.azure-pipelines/azure-pipelines-win.yml b/.azure-pipelines/azure-pipelines-win.yml
diff --git a/.ci_support/win_64_cuda_compilercuda-nvcccuda_compiler_version12.0.yaml b/.ci_support/win_64_cuda_compilercuda-nvcccuda_compiler_version12.0.yaml
@@ -0,0 +1,17 @@
+c_compiler:
+- vs2019
+channel_sources:
+- conda-forge
+channel_targets:
+- conda-forge main
+cuda_compiler:
+- cuda-nvcc
+cuda_compiler_version:
+- '12.0'
+cxx_compiler:
+- vs2019
+target_platform:
+- win-64
+zip_keys:
+- - cuda_compiler
+  - cuda_compiler_version
diff --git a/.ci_support/win_64_cuda_compilernvcccuda_compiler_version11.2.yaml b/.ci_support/win_64_cuda_compilernvcccuda_compiler_version11.2.yaml
@@ -0,0 +1,17 @@
+c_compiler:
+- vs2019
+channel_sources:
+- conda-forge
+channel_targets:
+- conda-forge main
+cuda_compiler:
+- nvcc
+cuda_compiler_version:
+- '11.2'
+cxx_compiler:
+- vs2019
+target_platform:
+- win-64
+zip_keys:
+- - cuda_compiler
+  - cuda_compiler_version
diff --git a/.ci_support/win_64_cuda_compilernvcccuda_compiler_version11.8.yaml b/.ci_support/win_64_cuda_compilernvcccuda_compiler_version11.8.yaml
@@ -0,0 +1,17 @@
+c_compiler:
+- vs2019
+channel_sources:
+- conda-forge
+channel_targets:
+- conda-forge main
+cuda_compiler:
+- nvcc
+cuda_compiler_version:
+- '11.8'
+cxx_compiler:
+- vs2019
+target_platform:
+- win-64
+zip_keys:
+- - cuda_compiler
+  - cuda_compiler_version
diff --git a/.scripts/run_win_build.bat b/.scripts/run_win_build.bat
diff --git a/README.md b/README.md
diff --git a/azure-pipelines.yml b/azure-pipelines.yml
diff --git a/recipe/meta.yaml b/recipe/meta.yaml
@@ -12,27 +12,34 @@ source:
     - osx-64-pick-discrete.patch  # [osx and x86_64]
 
 build:
-  skip: True  # [win]
-  skip: True  # [linux and cuda_compiler_version == "None"]
-  number: 4
+  skip: True  # [(linux or win) and cuda_compiler_version == "None"]
+  number: 5
   string: cuda{{ cuda_compiler_version | replace('.', '') }}_h{{ PKG_HASH }}_{{ PKG_BUILDNUM }}  # [cuda_compiler_version != "None"]
   string: cpu_h{{ PKG_HASH }}_{{ PKG_BUILDNUM }}                                                 # [cuda_compiler_version == "None"]
   string: mps_h{{ PKG_HASH }}_{{ PKG_BUILDNUM }}                                                 # [osx]
 
   script:
+    - LLAMA_ARGS="-DLLAMA_BUILD_TESTS=OFF"              # [unix]
+    - set LLAMA_ARGS=-DLLAMA_BUILD_TESTS=OFF           # [win]
     {% macro llama_args(value) -%}
-    - LLAMA_ARGS="${LLAMA_ARGS:-} -DLLAMA_{{ value }}"
+    - LLAMA_ARGS="${LLAMA_ARGS} -DLLAMA_{{ value }}"     # [unix]
+    - set LLAMA_ARGS=%LLAMA_ARGS% -DLLAMA_{{ value }}  # [win]
     {%- endmacro %}
-    {{ llama_args("BUILD_TESTS=OFF") }}
-    {{ llama_args("NATIVE=OFF") }}      # [osx and arm64]
+
+    {{ llama_args("NATIVE=OFF") }}      # [(osx and arm64) or win]
     {{ llama_args("AVX=OFF") }}         # [osx and arm64]
     {{ llama_args("AVX2=OFF") }}        # [osx and arm64]
     {{ llama_args("FMA=OFF") }}         # [osx and arm64]
     {{ llama_args("F16C=OFF") }}        # [osx and arm64]
     {{ llama_args("ACCELERATE=ON") }}   # [osx]
     {{ llama_args("METAL=ON") }}        # [osx]
     {{ llama_args("CUBLAS=ON") }}       # [cuda_compiler_version != "None"]
-    - cmake -S . -B build -G Ninja ${LLAMA_ARGS}
+
+    - echo $LLAMA_ARGS  # [unix]
+    - set LLAMA_ARGS    # [win]
+
+    - cmake -S . -B build -G Ninja %LLAMA_ARGS%   # [win]
+    - cmake -S . -B build -G Ninja ${LLAMA_ARGS}  # [unix]
     - cmake --build build --config Release
     - cmake --install build --prefix={{ PREFIX }}
 
@@ -49,6 +56,8 @@ requirements:
     - libcublas-dev    # [(cuda_compiler_version or "").startswith("12")]
   run:
     - cuda-version {{ cuda_compiler_version }}  # [cuda_compiler_version != "None"]
+    - cuda-cudart  {{ cuda_compiler_version }}  # [(cuda_compiler_version or "").startswith("12")]
+    - cuda-cudart_{{ target_platform }} {{ cuda_compiler_version }}  # [(cuda_compiler_version or "").startswith("12") and win]
 
 test:
   commands: