From 22007af045c99f74d8659ad99c931a5b47372da1 Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Tue, 17 Dec 2019 17:28:32 +0100
Subject: [PATCH] Do not link against cudart as default
---
src/pystencils_autodiff/tensorflow_jit.py | 9 +++++----
1 file changed, 5 insertions(+), 4 deletions(-)
diff --git a/src/pystencils_autodiff/tensorflow_jit.py b/src/pystencils_autodiff/tensorflow_jit.py
index 6a79443..d223a50 100644
--- a/src/pystencils_autodiff/tensorflow_jit.py
+++ b/src/pystencils_autodiff/tensorflow_jit.py
@@ -62,7 +62,7 @@ def link(object_files,
destination_file=None,
overwrite_destination_file=True,
additional_link_flags=[],
- link_cudart=True):
+ link_cudart=False):
"""Compiles given :param:`source_file` to a Tensorflow shared Library.
.. warning::
@@ -100,7 +100,7 @@ def link_and_load(object_files,
destination_file=None,
overwrite_destination_file=True,
additional_link_flags=[],
- link_cudart=True):
+ link_cudart=False):
import tensorflow as tf
destination_file = link(object_files,
@@ -190,7 +190,8 @@ def compile_sources_and_load(host_sources,
cuda_sources=[],
additional_compile_flags=[],
additional_link_flags=[],
- compile_only=False):
+ compile_only=False,
+ link_cudart=False):
import tensorflow as tf
@@ -226,7 +227,7 @@ def compile_sources_and_load(host_sources,
module_file = link(object_files,
overwrite_destination_file=False,
additional_link_flags=additional_link_flags,
- link_cudart=cuda_sources)
+ link_cudart=link_cudart and cuda_sources)
if not compile_only:
module = tf.load_op_library(module_file)
if module:
--
GitLab