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";