Commit c6e5fda1 authored by Stephan Seitz's avatar Stephan Seitz
Browse files

Fixup

parent aa5f89f0
......@@ -123,7 +123,7 @@ class TorchModule(JinjaCppFile):
[file_name],
with_cuda=self.is_cuda,
extra_cflags=['--std=c++14'],
extra_cuda_cflags=['-std=c++14', '--ccbin', get_compiler_config()['command']],
extra_cuda_cflags=['-std=c++14', '-ccbin', get_compiler_config()['command']],
build_directory=build_dir,
extra_include_paths=[get_pycuda_include_path(),
get_pystencils_include_path()])
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment