Add CUDA support for aarch64/ppc64le

conda-forge · Oct 11, 2021 · 94628f1 · 94628f1
1 parent 4eddb94
commit 94628f1
Show file tree

Hide file tree

Showing 3 changed files with 43 additions and 22 deletions.
diff --git a/recipe/conda_build_config.yaml b/recipe/conda_build_config.yaml
@@ -116,6 +116,8 @@ cdt_name:  # [linux]
 
   - cos6   # [linux64 and os.environ.get("DEFAULT_LINUX_VERSION", "cos6") == "cos6"]
   - cos7   # [linux64 and os.environ.get("DEFAULT_LINUX_VERSION", "cos6") == "cos7"]
+  - cos7   # [linux64 and ppc64le and os.environ.get("DEFAULT_LINUX_VERSION", "cos6") == "cos7"]
+  - cos7   # [linux64 and aarch64 and os.environ.get("DEFAULT_LINUX_VERSION", "cos6") == "cos7"]
 
 docker_image:                                   # [os.environ.get("BUILD_PLATFORM", "").startswith("linux-")]
   - quay.io/condaforge/linux-anvil-comp7        # [os.environ.get("BUILD_PLATFORM") == "linux-64" and os.environ.get("DEFAULT_LINUX_VERSION", "cos6") == "cos6"]
@@ -126,6 +128,8 @@ docker_image:                                   # [os.environ.get("BUILD_PLATFOR
 
   - quay.io/condaforge/linux-anvil-cuda:10.2       # [linux64 and os.environ.get("BUILD_PLATFORM") == "linux-64" and os.environ.get("DEFAULT_LINUX_VERSION", "cos6") == "cos6"]
   - quay.io/condaforge/linux-anvil-cos7-cuda:10.2  # [linux64 and os.environ.get("BUILD_PLATFORM") == "linux-64" and os.environ.get("DEFAULT_LINUX_VERSION", "cos6") == "cos7"]
+  - quay.io/condaforge/linux-anvil-ppc64le-cuda:10.2  # [linux and ppc64le and os.environ.get("DEFAULT_LINUX_VERSION", "cos6") == "cos7"]
+  - quay.io/condaforge/linux-anvil-aarch64-cuda:10.2  # [linux and aarch64 and os.environ.get("DEFAULT_LINUX_VERSION", "cos6") == "cos7"]
 
 zip_keys:
   -                             # [unix]
@@ -136,11 +140,11 @@ zip_keys:
     - python
     - numpy
     - python_impl
-  -                             # [linux64]
-    - cudnn                     # [linux64]
-    - cuda_compiler_version     # [linux64]
-    - cdt_name                  # [linux64]
-    - docker_image              # [linux64 and os.environ.get("BUILD_PLATFORM") == "linux-64"]
+  -                             # [linux]
+    - cudnn                     # [linux]
+    - cuda_compiler_version     # [linux]
+    - cdt_name                  # [linux]
+    - docker_image              # [linux and os.environ.get("BUILD_PLATFORM") == "linux-64"]
 
 # aarch64 specifics because conda-build sets many things to centos 6
 # this can probably be removed when conda-build gets updated defaults

diff --git a/recipe/migrations/cuda110.yaml b/recipe/migrations/cuda110.yaml
@@ -31,14 +31,22 @@ __migrator:
       - quay.io/condaforge/linux-anvil-cos7-cuda:10.1    # [linux64 and os.environ.get("BUILD_PLATFORM") == "linux-64"]
       - quay.io/condaforge/linux-anvil-cuda:10.2    # [linux64 and os.environ.get("BUILD_PLATFORM") == "linux-64"]
       - quay.io/condaforge/linux-anvil-cos7-cuda:10.2    # [linux64 and os.environ.get("BUILD_PLATFORM") == "linux-64"]
+      - quay.io/condaforge/linux-anvil-ppc64le-cuda:10.2  # [linux and ppc64le and os.environ.get("DEFAULT_LINUX_VERSION", "cos6") == "cos7"]
+      - quay.io/condaforge/linux-anvil-aarch64-cuda:10.2  # [linux and aarch64 and os.environ.get("DEFAULT_LINUX_VERSION", "cos6") == "cos7"]
       - quay.io/condaforge/linux-anvil-cuda:11.0    # [linux64 and os.environ.get("BUILD_PLATFORM") == "linux-64"]
+      - quay.io/condaforge/linux-anvil-ppc64le-cuda:11.0  # [linux and ppc64le and os.environ.get("DEFAULT_LINUX_VERSION", "cos6") == "cos7"]
+      - quay.io/condaforge/linux-anvil-aarch64-cuda:11.0  # [linux and aarch64 and os.environ.get("DEFAULT_LINUX_VERSION", "cos6") == "cos7"]
       - quay.io/condaforge/linux-anvil-cuda:11.1    # [linux64 and os.environ.get("BUILD_PLATFORM") == "linux-64"]
+      - quay.io/condaforge/linux-anvil-ppc64le-cuda:11.1  # [linux and ppc64le and os.environ.get("DEFAULT_LINUX_VERSION", "cos6") == "cos7"]
+      - quay.io/condaforge/linux-anvil-aarch64-cuda:11.1  # [linux and aarch64 and os.environ.get("DEFAULT_LINUX_VERSION", "cos6") == "cos7"]
       - quay.io/condaforge/linux-anvil-cuda:11.2    # [linux64 and os.environ.get("BUILD_PLATFORM") == "linux-64"]
+      - quay.io/condaforge/linux-anvil-ppc64le-cuda:11.2  # [linux and ppc64le and os.environ.get("DEFAULT_LINUX_VERSION", "cos6") == "cos7"]
+      - quay.io/condaforge/linux-anvil-aarch64-cuda:11.2  # [linux and aarch64 and os.environ.get("DEFAULT_LINUX_VERSION", "cos6") == "cos7"]
 
 cuda_compiler_version:
   - None
-  - 10.2                       # [linux64]
-  - 11.0                       # [linux64]
+  - 10.2                       # [linux]
+  - 11.0                       # [linux]
 
 c_compiler_version:     # [linux]
   - 7                   # [linux64 or aarch64]
@@ -49,8 +57,8 @@ cxx_compiler_version:   # [linux]
 
 cudnn:
   - undefined
-  - 7                   # [linux64]
-  - 8                   # [linux64]
+  - 7                   # [linux]
+  - 8                   # [linux]
 
 cdt_name:  # [linux]
   - cos6   # [linux64 and os.environ.get("DEFAULT_LINUX_VERSION", "cos6") == "cos6"]
@@ -61,7 +69,8 @@ cdt_name:  # [linux]
 
   - cos6   # [linux64 and os.environ.get("DEFAULT_LINUX_VERSION", "cos6") == "cos6"]
   - cos7   # [linux64 and os.environ.get("DEFAULT_LINUX_VERSION", "cos6") == "cos7"]
-  - cos7   # [linux64]
+  - cos7   # [linux and ppc64le]
+  - cos7   # [linux and aarch64]
 
 docker_image:                                   # [os.environ.get("BUILD_PLATFORM", "").startswith("linux-")]
   - quay.io/condaforge/linux-anvil-comp7        # [os.environ.get("BUILD_PLATFORM") == "linux-64" and os.environ.get("DEFAULT_LINUX_VERSION", "cos6") == "cos6"]
@@ -72,10 +81,14 @@ docker_image:                                   # [os.environ.get("BUILD_PLATFOR
 
   - quay.io/condaforge/linux-anvil-cuda:10.2        # [linux64 and os.environ.get("BUILD_PLATFORM") == "linux-64" and os.environ.get("DEFAULT_LINUX_VERSION", "cos6") == "cos6"]
   - quay.io/condaforge/linux-anvil-cos7-cuda:10.2   # [linux64 and os.environ.get("BUILD_PLATFORM") == "linux-64" and os.environ.get("DEFAULT_LINUX_VERSION", "cos6") == "cos7"]
+  - quay.io/condaforge/linux-anvil-ppc64le-cuda:10.2  # [linux and ppc64le and os.environ.get("DEFAULT_LINUX_VERSION", "cos6") == "cos7"]
+  - quay.io/condaforge/linux-anvil-aarch64-cuda:10.2  # [linux and aarch64 and os.environ.get("DEFAULT_LINUX_VERSION", "cos6") == "cos7"]
   - quay.io/condaforge/linux-anvil-cuda:11.0    # [linux64 and os.environ.get("BUILD_PLATFORM") == "linux-64"]
+  - quay.io/condaforge/linux-anvil-ppc64le-cuda:11.0  # [linux and ppc64le and os.environ.get("DEFAULT_LINUX_VERSION", "cos6") == "cos7"]
+  - quay.io/condaforge/linux-anvil-aarch64-cuda:11.0  # [linux and aarch64 and os.environ.get("DEFAULT_LINUX_VERSION", "cos6") == "cos7"]
 
 zip_keys:
-  - - cudnn                      # [linux64]
-    - cuda_compiler_version      # [linux64]
-    - docker_image               # [linux64 and os.environ.get("BUILD_PLATFORM") == "linux-64"]
-    - cdt_name                   # [linux64]
+  - - cudnn                      # [linux]
+    - cuda_compiler_version      # [linux]
+    - docker_image               # [linux and os.environ.get("BUILD_PLATFORM") == "linux-64"]
+    - cdt_name                   # [linux]
diff --git a/recipe/migrations/cuda111_112.yaml b/recipe/migrations/cuda111_112.yaml
@@ -12,18 +12,22 @@ __migrator:
   check_solvable: false
   primary_key: cuda_compiler_version
 
-cuda_compiler_version:         # [linux64 or win]
-  - 11.1                       # [linux64 or win]
-  - 11.2                       # [linux64 or win]
+cuda_compiler_version:         # [linux or win]
+  - 11.1                       # [linux or win]
+  - 11.2                       # [linux or win]
 
-cudnn:                  # [linux64 or win]
-  - 8                   # [linux64 or win]
-  - 8                   # [linux64 or win]
+cudnn:                  # [linux or win]
+  - 8                   # [linux or win]
+  - 8                   # [linux or win]
 
 cdt_name:  # [linux]
-  - cos7   # [linux64]
-  - cos7   # [linux64]
+  - cos7   # [linux]
+  - cos7   # [linux]
 
 docker_image:                                   # [os.environ.get("BUILD_PLATFORM", "").startswith("linux-")]
   - quay.io/condaforge/linux-anvil-cuda:11.1    # [linux64 and os.environ.get("BUILD_PLATFORM") == "linux-64"]
+  - quay.io/condaforge/linux-anvil-ppc64le-cuda:11.1  # [linux and ppc64le and os.environ.get("DEFAULT_LINUX_VERSION", "cos6") == "cos7"]
+  - quay.io/condaforge/linux-anvil-aarch64-cuda:11.1  # [linux and aarch64 and os.environ.get("DEFAULT_LINUX_VERSION", "cos6") == "cos7"]
   - quay.io/condaforge/linux-anvil-cuda:11.2    # [linux64 and os.environ.get("BUILD_PLATFORM") == "linux-64"]
+  - quay.io/condaforge/linux-anvil-ppc64le-cuda:11.2  # [linux and ppc64le and os.environ.get("DEFAULT_LINUX_VERSION", "cos6") == "cos7"]
+  - quay.io/condaforge/linux-anvil-aarch64-cuda:11.2  # [linux and aarch64 and os.environ.get("DEFAULT_LINUX_VERSION", "cos6") == "cos7"]