Use conda compilers

Signed-off-by: Jordan Jacobelli <jjacobelli@nvidia.com>
rapidsai · Mar 14, 2022 · db728d9 · db728d9
1 parent f31cd8b
commit db728d9
Show file tree

Hide file tree

Showing 4 changed files with 47 additions and 11 deletions.
diff --git a/conda/recipes/librmm/conda_build_config.yaml b/conda/recipes/librmm/conda_build_config.yaml
@@ -0,0 +1,6 @@
+c_compiler_version:       # [linux]
+  - 9                     # [linux]
+cxx_compiler_version:     # [linux]
+  - 9                     # [linux]
+fortran_compiler_version: # [linux]
+  - 9                     # [linux]
diff --git a/conda/recipes/librmm/meta.yaml b/conda/recipes/librmm/meta.yaml
@@ -20,9 +20,6 @@ outputs:
       number: {{ GIT_DESCRIBE_NUMBER }}
       string: cuda{{ cuda_major }}_{{ GIT_DESCRIBE_HASH }}_{{ GIT_DESCRIBE_NUMBER }}
       script_env: &script_env
-        - CC
-        - CXX
-        - CUDAHOSTCXX
         - PARALLEL_LEVEL
         - CMAKE_GENERATOR
         - CMAKE_C_COMPILER_LAUNCHER
@@ -35,9 +32,17 @@ outputs:
         - 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 {{ cmake_version }}
+        - {{ compiler('cxx') }}
+        - 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 }}.*
       run:
@@ -88,9 +93,17 @@ outputs:
       number: {{ GIT_DESCRIBE_NUMBER }}
       string: cuda{{ cuda_major }}_{{ GIT_DESCRIBE_HASH }}_{{ GIT_DESCRIBE_NUMBER }}
       script_env: *script_env
+      ignore_run_exports_from:
+        - nvcc_linux-64 # [linux64]
+        - nvcc_linux-aarch64 # [aarch64]
     requirements:
       build:
         - cmake {{ cmake_version }}
+        - {{ compiler('cxx') }}
+        - 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 }}.*
       run:

diff --git a/conda/recipes/rmm/conda_build_config.yaml b/conda/recipes/rmm/conda_build_config.yaml
@@ -1,3 +1,13 @@
 cudaMallocAsync:
   - has_cma
   - no_cma
+
+c_compiler_version:       # [linux]
+  - 9                     # [linux]
+cxx_compiler_version:     # [linux]
+  - 9                     # [linux]
+fortran_compiler_version: # [linux]
+  - 9                     # [linux]
+
+cuda_compiler:
+  - nvcc
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') }}
+    - {{ compiler('cxx') }}
+    - 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:
+    - cuda-python >=11.5,<12.0
+    - cudatoolkit {{ cuda_version }}.*
+    - cython >=0.29,<0.30
+    - librmm {{ version }}.*
     - python
-    - librmm {{ version }}
     - 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 %}