From 797bf4bd21aed94e578769e3f4c2bc3a26c5e364 Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Fri, 29 Nov 2019 17:46:39 +0100
Subject: [PATCH] Fix unused parameter
---
src/pystencils_autodiff/tensorflow_jit.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/pystencils_autodiff/tensorflow_jit.py b/src/pystencils_autodiff/tensorflow_jit.py
index 9e58073..97f10e0 100644
--- a/src/pystencils_autodiff/tensorflow_jit.py
+++ b/src/pystencils_autodiff/tensorflow_jit.py
@@ -139,7 +139,7 @@ if pystencils.gpucuda.cudajit.USE_FAST_MATH:
def compile_file(file,
use_nvcc=False,
- nvcc='nvcc',
+ nvcc=None,
overwrite_destination_file=True,
additional_compile_flags=[],
openmp=True):
@@ -148,7 +148,7 @@ def compile_file(file,
import tensorflow as tf
if use_nvcc:
- command_prefix = [NVCC_BINARY,
+ command_prefix = [nvcc or NVCC_BINARY,
'--expt-relaxed-constexpr',
'-ccbin',
get_compiler_config()['tensorflow_host_compiler'],
--
GitLab