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

Lint

parent 74d93c0e
No related branches found
No related tags found
No related merge requests found
Pipeline #20817 failed
...@@ -206,11 +206,12 @@ setup_pybind11(cfg) ...@@ -206,11 +206,12 @@ setup_pybind11(cfg)
except Exception as e: except Exception as e:
print(e) print(e)
torch_extension = load(hash, torch_extension = load(hash,
[file_name], [file_name],
with_cuda=self.is_cuda, with_cuda=self.is_cuda,
extra_cflags=['--std=c++14', get_compiler_config()['flags'].replace('--std=c++11', '')], extra_cflags=['--std=c++14',
extra_cuda_cflags=['-std=c++14', '-ccbin', get_compiler_config()['command']], get_compiler_config()['flags'].replace('--std=c++11', '')],
build_directory=cache_dir, extra_cuda_cflags=['-std=c++14', '-ccbin', get_compiler_config()['command']],
extra_include_paths=[get_pycuda_include_path(), build_directory=cache_dir,
get_pystencils_include_path()]) extra_include_paths=[get_pycuda_include_path(),
get_pystencils_include_path()])
return torch_extension return torch_extension
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment