From e67cc03e28eba81e167d1c5380da495146868d5d Mon Sep 17 00:00:00 2001 From: John Kirkham Date: Thu, 10 Oct 2019 13:56:42 -0400 Subject: [PATCH] Add the CUDA compiler --- recipe/conda_build_config.yaml | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/recipe/conda_build_config.yaml b/recipe/conda_build_config.yaml index 61c351a7360..85b2d90cfb9 100644 --- a/recipe/conda_build_config.yaml +++ b/recipe/conda_build_config.yaml @@ -40,6 +40,15 @@ m2w64_fortran_compiler: # [win] - m2w64-toolchain # [win] CMAKE_GENERATOR: # [win] - NMake Makefiles # [win] + +cuda_compiler: # [linux64] + - nvcc # [linux64] +cuda_compiler_version: # [linux64] + - None # [linux64] + - 9.2 # [linux64] + - 10.0 # [linux64] + - 10.1 # [linux64] + # # Go Compiler Options # @@ -96,6 +105,9 @@ channel_targets: docker_image: # [linux] - condaforge/linux-anvil-comp7 # [linux64] + - condaforge/linux-anvil-cuda:9.2 # [linux64] + - condaforge/linux-anvil-cuda:10.0 # [linux64] + - condaforge/linux-anvil-cuda:10.1 # [linux64] - condaforge/linux-anvil-aarch64 # [aarch64] - condaforge/linux-anvil-ppc64le # [ppc64le] @@ -109,6 +121,9 @@ zip_keys: - vc # [win] - c_compiler # [win] - cxx_compiler # [win] + - + - cuda_compiler_version # [linux64] + - docker_image # [linux64] # aarch64 specifics because conda-build sets many things to centos 6 # this can probably be removed when conda-build gets updated defaults