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

Alright, -DGOOGLE_CUDA is still required

parent 23f0b9b8
No related merge requests found
......@@ -28,7 +28,7 @@ assert _pyronn_layers_sources, "Could not find the source files of PYRO-NN-Layer
_pyronn_layers_module_file = pystencils_autodiff.tensorflow_jit.compile_sources_and_load(
[],
_pyronn_layers_sources,
additional_compile_flags=['-I' + _pyronn_layers_dir],
additional_compile_flags=['-I' + _pyronn_layers_dir, '-DGOOGLE_CUDA=1'],
additional_link_flags=['-lcudart', '-L/usr/local/cuda/lib64'],
compile_only=True,
link_cudart=True) # TODO: msvc! \D
......
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