Merge pull request #30 from adibbley/add-activation

Add windows activation scripts
conda-forge · Nov 8, 2023 · 394f381 · 394f381
2 parents cbf8f7c + 3c74590
commit 394f381
Show file tree

Hide file tree

Showing 5 changed files with 39 additions and 4 deletions.
diff --git a/recipe/activate.bat b/recipe/activate.bat
@@ -0,0 +1,9 @@
+@echo on
+
+:: Backup environment variables (only if the variables are set)
+if defined INCLUDE (
+    set "INCLUDE_CONDA_NVCC_BACKUP=%INCLUDE%"
+)
+
+:: Append `targets` to search path to give exist includes preference
+set "INCLUDE=%INCLUDE%;%LIBRARY_INC%\targets\x64"
diff --git a/recipe/bld.bat b/recipe/bld.bat
@@ -0,0 +1,16 @@
+@echo on
+setlocal enableextensions enabledelayedexpansion || goto :error
+
+:: Activation script
+mkdir %PREFIX%\etc\conda\activate.d
+copy %RECIPE_DIR%\activate.bat %PREFIX%\etc\conda\activate.d\~cuda-nvcc_activate.bat || goto :error
+
+:: Deactivation script
+mkdir %PREFIX%\etc\conda\deactivate.d
+copy %RECIPE_DIR%\deactivate.bat %PREFIX%\etc\conda\deactivate.d\~cuda-nvcc_deactivate.bat || goto :error
+
+goto :EOF
+
+:error
+echo Failed with error %errorlevel%
+exit /b %errorlevel%
diff --git a/recipe/deactivate.bat b/recipe/deactivate.bat
@@ -0,0 +1,7 @@
+@echo on
+
+:: Restore environment variables (if there is anything to restore)
+if defined INCLUDE_CONDA_NVCC_BACKUP (
+	set "INCLUDE=%INCLUDE_CONDA_NVCC_BACKUP%"
+	set "INCLUDE_CONDA_NVCC_BACKUP="
+)
diff --git a/recipe/meta.yaml b/recipe/meta.yaml
@@ -12,7 +12,7 @@ source:
   sha256: 5db25d4fd138013b563f9a3d1d87f7de7df1dac014fd4cccdfbb3435a5cff761
 
 build:
-  number: 11
+  number: 12
   skip: true  # [osx]
   skip: true  # [target_platform != "linux-64" and target_platform != cross_target_platform]
 
@@ -43,9 +43,11 @@ outputs:
     run_exports:
       strong:
         - cuda-version >={{ cuda_version }},<{{ cuda_version.split(".")[0]|int + 1 }}
-    files:                                               # [linux]
-      - etc/conda/activate.d/~cuda-nvcc_activate.sh      # [linux]
-      - etc/conda/deactivate.d/~cuda-nvcc_deactivate.sh  # [linux]
+    files:
+      - etc/conda/activate.d/~cuda-nvcc_activate.sh       # [linux]
+      - etc/conda/deactivate.d/~cuda-nvcc_deactivate.sh   # [linux]
+      - etc/conda/activate.d/~cuda-nvcc_activate.bat      # [win]
+      - etc/conda/deactivate.d/~cuda-nvcc_deactivate.bat  # [win]
     requirements:
       host:
         - sysroot_{{ cross_target_platform }} 2.17.*  # [linux]

diff --git a/recipe/test.cpp b/recipe/test.cpp
@@ -1,4 +1,5 @@
 #include <cuda_runtime.h>
+#include <thrust/version.h>
 
 int main() {
     return 0;