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

Fix unused parameter

parent 3b21e27a
Branches
Tags
No related merge requests found
Hide whitespace changes
Inline Side-by-side
src/pystencils_autodiff/tensorflow_jit.py
+ 2
− 2
View file @ 797bf4bd
...@@ -139,7 +139,7 @@ if pystencils.gpucuda.cudajit.USE_FAST_MATH: ...@@ -139,7 +139,7 @@ if pystencils.gpucuda.cudajit.USE_FAST_MATH:
def compile_file(file, def compile_file(file,
use_nvcc=False, use_nvcc=False,
nvcc='nvcc', nvcc=None,
overwrite_destination_file=True, overwrite_destination_file=True,
additional_compile_flags=[], additional_compile_flags=[],
openmp=True): openmp=True):
...@@ -148,7 +148,7 @@ def compile_file(file, ...@@ -148,7 +148,7 @@ def compile_file(file,
import tensorflow as tf import tensorflow as tf
if use_nvcc: if use_nvcc:
command_prefix = [NVCC_BINARY, command_prefix = [nvcc or NVCC_BINARY,
'--expt-relaxed-constexpr', '--expt-relaxed-constexpr',
'-ccbin', '-ccbin',
get_compiler_config()['tensorflow_host_compiler'], get_compiler_config()['tensorflow_host_compiler'],
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment