From 4bc28e09e13e32127f480f0517c9b01ffb4ff0aa Mon Sep 17 00:00:00 2001
From: Mark Glines <mark@glines.org>
Date: Wed, 4 Nov 2020 11:19:22 -0500
Subject: [PATCH] Add TACO_NVCC var to complement TACO_NVCCFLAGS

This is useful for passing specific arguments to nvcc.  In my case,
I wanted to force nvcc to use a specific version of g++.
---
 src/codegen/module.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/codegen/module.cpp b/src/codegen/module.cpp
index adeecbe76..9d99c16f7 100644
--- a/src/codegen/module.cpp
+++ b/src/codegen/module.cpp
@@ -122,7 +122,7 @@ string Module::compile() {
   string file_ending;
   string shims_file;
   if (should_use_CUDA_codegen()) {
-    cc = "nvcc";
+    cc = util::getFromEnv("TACO_NVCC", "nvcc");
     cflags = util::getFromEnv("TACO_NVCCFLAGS",
     get_default_CUDA_compiler_flags());
     file_ending = ".cu";