Skip to content

Commit

Permalink
Add TACO_NVCC var to complement TACO_NVCCFLAGS
Browse files Browse the repository at this point in the history
This is useful for passing specific arguments to nvcc.  In my case,
I wanted to force nvcc to use a specific version of g++.
  • Loading branch information
Infinoid committed Nov 4, 2020
1 parent 7441f3a commit 4bc28e0
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/codegen/module.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -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";
Expand Down

0 comments on commit 4bc28e0

Please sign in to comment.