Use conda compilers

Signed-off-by: Jordan Jacobelli <jjacobelli@nvidia.com>
rapidsai · Mar 2, 2022 · 5d009e0 · 5d009e0
1 parent 514f4b4
commit 5d009e0
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 12 deletions.
diff --git a/conda/recipes/librmm/meta.yaml b/conda/recipes/librmm/meta.yaml
@@ -15,9 +15,6 @@ build:
   number: {{ GIT_DESCRIBE_NUMBER }}
   string: cuda{{ cuda_major }}_{{ GIT_DESCRIBE_HASH }}_{{ GIT_DESCRIBE_NUMBER }}
   script_env:
-    - CC
-    - CXX
-    - CUDAHOSTCXX
     - PARALLEL_LEVEL
     - VERSION_SUFFIX
     - CMAKE_GENERATOR
@@ -32,12 +29,20 @@ build:
     - SCCACHE_IDLE_TIMEOUT=32768
   run_exports:
     - {{ pin_subpackage("librmm", max_pin="x.x") }}
+  ignore_run_exports_from:
+    - nvcc_linux-64 # [linux64]
+    - nvcc_linux-aarch64 # [aarch64]
 
 requirements:
   build:
     - cmake >=3.20.1
+    - {{ compiler('cxx') }} 9
+    - nvcc_linux-64 {{ cuda_version }} # [linux64]
+    - nvcc_linux-aarch64 {{ cuda_version }} # [aarch64]
+    - sysroot_linux-64 2.17  # [linux64]
+    - sysroot_linux-aarch64 2.17  # [aarch64]
   host:
-    - cudatoolkit {{ cuda_version }}.*
+    - cudatoolkit {{ cuda_version }}
   run:
     - spdlog>=1.8.5,<1.9
     - {{ pin_compatible('cudatoolkit', max_pin='x', min_pin='x') }}

diff --git a/conda/recipes/rmm/meta.yaml b/conda/recipes/rmm/meta.yaml
@@ -18,23 +18,30 @@ build:
   script_env:
     - RMM_BUILD_NO_GPU_TEST
     - VERSION_SUFFIX
-    - CC
-    - CXX
-    - CUDAHOSTCXX
+  ignore_run_exports_from:
+    - nvcc_linux-64 # [linux64]
+    - nvcc_linux-aarch64 # [aarch64]
 
 requirements:
+  build:
+    - {{ compiler('c') }} 9
+    - {{ compiler('cxx') }} 9
+    - sysroot_linux-64 2.17  # [linux64]
+    - sysroot_linux-aarch64 2.17  # [aarch64]
+    - nvcc_linux-64 {{ cuda_version }} # [linux64]
+    - nvcc_linux-aarch64 {{ cuda_version }} # [aarch64]
   host:
-    - python
+    - cuda-python >=11.5,<12.0
+    - cudatoolkit {{ cuda_version }}
+    - cython >=0.29,<0.30
     - librmm {{ version }}
+    - python
     - setuptools
-    - cython >=0.29,<0.30
     - spdlog>=1.8.5,<2.0.0a0
-    - cudatoolkit {{ cuda_version }}.*
-    - cuda-python >=11.5,<12.0
   run:
-    - python
     - numba >=0.49
     - numpy
+    - python
 {% if cudaMallocAsync == "has_cma" %}
     - {{ pin_compatible('cudatoolkit', max_pin='x', lower_bound='11.2') }} # cudatoolkit >=11.2,<12.0.0
 {% else %}