Merge pull request tensor-compiler#321 from Infinoid/taco_nvcc

Add TACO_NVCC var to complement TACO_NVCCFLAGS
rohany · Dec 16, 2020 · 81ebb64 · 81ebb64
2 parents dcbfdc3 + 4bc28e0
commit 81ebb64
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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";