Shim package to interact with an **existing** NVCC install

recipes/nvcc/LICENSE.txt

@@ -0,0 +1,26 @@
+Copyright (c) 2019, NVIDIA CORPORATION. All rights reserved.
+
+Redistribution and use in source and binary forms, with or without
+modification, are permitted provided that the following conditions
+are met:
+
+ * Redistributions of source code must retain the above copyright
+   notice, this list of conditions and the following disclaimer.
+ * Redistributions in binary form must reproduce the above copyright
+   notice, this list of conditions and the following disclaimer in the
+   documentation and/or other materials provided with the distribution.
+ * Neither the name of NVIDIA CORPORATION nor the names of its
+   contributors may be used to endorse or promote products derived
+   from this software without specific prior written permission.
+
+THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS ``AS IS'' AND ANY
+EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
+PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE COPYRIGHT OWNER OR
+CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
+EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
+PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
+PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
+OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

recipes/nvcc/build.sh

@@ -0,0 +1,34 @@
+#!/bin/bash
+
+set -xeuo pipefail
+
+# Default to using `nvcc` to specify `CUDA_HOME`.
+if [ -z ${CUDA_HOME+x} ]
+then
+    CUDA_HOME="$(dirname $(dirname $(which nvcc)))"
+fi
+
+# Set `CUDA_HOME` in an activation script.
+mkdir -p "${PREFIX}/etc/conda/activate.d"
+cat > "${PREFIX}/etc/conda/activate.d/${PKG_NAME}_activate.sh" <<EOF
+#!/bin/bash
+export CUDA_HOME="${CUDA_HOME}"
+EOF
+
+# Unset `CUDA_HOME` in a deactivation script.
+mkdir -p "${PREFIX}/etc/conda/deactivate.d"
+cat > "${PREFIX}/etc/conda/deactivate.d/${PKG_NAME}_deactivate.sh" <<EOF
+#!/bin/bash
+unset CUDA_HOME
+EOF
+
+# Symlink `nvcc` into `bin` so it can be easily run.
+mkdir -p "${PREFIX}/bin"
+ln -s "${CUDA_HOME}/bin/nvcc" "${PREFIX}/bin/nvcc"
+
+# Add `libcuda.so` shared object stub to the compiler sysroot.
+# Needed for things that want to link to `libcuda.so`.
+# Stub is used to avoid getting driver code linked into binaries.
+CONDA_BUILD_SYSROOT="$(${CC} --print-sysroot)"
+mkdir -p "${CONDA_BUILD_SYSROOT}/lib"
+ln -s "${CUDA_HOME}/lib64/stubs/libcuda.so" "${CONDA_BUILD_SYSROOT}/lib/libcuda.so"

recipes/nvcc/meta.yaml

@@ -0,0 +1,51 @@
+{% set name = "nvcc" %}
+{% set version = "9.2" %}
+
+package:
+  name: "{{ name }}_{{ target_platform }}"
+  version: {{ version }}
+
+build:
+  number: 0
+  skip: true  # [not linux]
+  script_env:
+    - CUDA_HOME
+  ignore_run_exports:
+    - libgcc-ng
+  run_exports:
+    strong:
+      - cudatoolkit {{ version }}|{{ version }}.*
+
+requirements:
+  host:
+    - {{ compiler("c") }}
+  run:
+    - {{ compiler("c") }}
+    # Host code is forwarded to a C++ compiler
+    - {{ compiler("cxx") }}
+
+test:
+  commands:
+    # Verify the symlink to the libcuda stub library exists.
+    - test -f "$(${CC} --print-sysroot)/lib/libcuda.so"
+
+    # Verify the activation scripts are in-place.
+    {% for state in ["activate", "deactivate"] %}
+    - test -f "${PREFIX}/etc/conda/{{ state }}.d/{{ PKG_NAME }}_{{ state }}.sh"
+    {% endfor %}
+
+    # Try using the activation scripts.
+    - source ${PREFIX}/etc/conda/activate.d/{{ PKG_NAME }}_activate.sh
+    - if [ -z ${CUDA_HOME+x} ]; then echo "CUDA_HOME is unset after activation" && exit 1; else echo "CUDA_HOME is set to '$CUDA_HOME'"; fi
+    - source ${PREFIX}/etc/conda/deactivate.d/{{ PKG_NAME }}_deactivate.sh
+    - if [ -z ${CUDA_HOME+x} ]; then echo "CUDA_HOME is unset after deactivation "; else echo "CUDA_HOME is set to '$CUDA_HOME' after deactivation" && exit 1; fi
+
+about:
+  home: https://github.com/conda-forge/nvcc-feedstock
+  license: BSD 3-Clause
+  license_file: LICENSE.txt
+  summary: A meta-package to enable the right nvcc.
+
+extra:
+  recipe-maintainers:
+    - jakirkham