From bc366f07981e93eae8b213a0777f7143507e2549 Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Thu, 12 Sep 2019 20:04:45 +0200
Subject: [PATCH] Lint
---
src/pystencils_autodiff/tensorflow_jit.py | 8 ++++++--
1 file changed, 6 insertions(+), 2 deletions(-)
diff --git a/src/pystencils_autodiff/tensorflow_jit.py b/src/pystencils_autodiff/tensorflow_jit.py
index 6b8fcae..05a1269 100644
--- a/src/pystencils_autodiff/tensorflow_jit.py
+++ b/src/pystencils_autodiff/tensorflow_jit.py
@@ -161,12 +161,16 @@ def compile_sources_and_load(host_sources, cuda_sources=[], additional_compile_f
file_name = join(pystencils.cache.cache_dir, f'{_hash(source_code.encode()).hexdigest()}{file_extension}')
write_file(file_name, source_code)
- compile_file(file_name, use_nvcc=is_cuda, overwrite_destination_file=False,
+ compile_file(file_name,
+ use_nvcc=is_cuda,
+ overwrite_destination_file=False,
additional_compile_flags=additional_compile_flags)
object_files.append(file_name + '.o')
print('Linking Tensorflow module...')
- module = link_and_load(object_files, overwrite_destination_file=False, additional_link_flags=additional_link_flags)
+ module = link_and_load(object_files,
+ overwrite_destination_file=False,
+ additional_link_flags=additional_link_flags)
if module:
print('Loaded Tensorflow module.')
return module
--
GitLab