diff --git a/src/pystencils_autodiff/backends/astnodes.py b/src/pystencils_autodiff/backends/astnodes.py index a84f7bfe93ac8ff4ae8dbb725bf25578c2871a95..286b39cf75bef980b745eec3a98734f9c70c9e23 100644 --- a/src/pystencils_autodiff/backends/astnodes.py +++ b/src/pystencils_autodiff/backends/astnodes.py @@ -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()])