Skip to content
Snippets Groups Projects
Commit c6e5fda1 authored by Stephan Seitz's avatar Stephan Seitz
Browse files

Fixup

parent aa5f89f0
No related branches found
No related tags found
No related merge requests found
...@@ -123,7 +123,7 @@ class TorchModule(JinjaCppFile): ...@@ -123,7 +123,7 @@ class TorchModule(JinjaCppFile):
[file_name], [file_name],
with_cuda=self.is_cuda, with_cuda=self.is_cuda,
extra_cflags=['--std=c++14'], 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, build_directory=build_dir,
extra_include_paths=[get_pycuda_include_path(), extra_include_paths=[get_pycuda_include_path(),
get_pystencils_include_path()]) get_pystencils_include_path()])
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment