Skip to content

Commit 81ebb64

Browse files
Merge pull request #321 from Infinoid/taco_nvcc
Add TACO_NVCC var to complement TACO_NVCCFLAGS
2 parents dcbfdc3 + 4bc28e0 commit 81ebb64

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/codegen/module.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -122,7 +122,7 @@ string Module::compile() {
122122
string file_ending;
123123
string shims_file;
124124
if (should_use_CUDA_codegen()) {
125-
cc = "nvcc";
125+
cc = util::getFromEnv("TACO_NVCC", "nvcc");
126126
cflags = util::getFromEnv("TACO_NVCCFLAGS",
127127
get_default_CUDA_compiler_flags());
128128
file_ending = ".cu";

0 commit comments

Comments
 (0)