From 3a3bd16ad5b8434c1690fc5d563f4aeec160e330 Mon Sep 17 00:00:00 2001 From: Stephan Seitz <stephan.seitz@fau.de> Date: Mon, 23 Sep 2019 14:05:44 +0200 Subject: [PATCH] Link cudart --- src/pystencils_autodiff/tensorflow_jit.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/pystencils_autodiff/tensorflow_jit.py b/src/pystencils_autodiff/tensorflow_jit.py index e5312b0..916912a 100644 --- a/src/pystencils_autodiff/tensorflow_jit.py +++ b/src/pystencils_autodiff/tensorflow_jit.py @@ -75,6 +75,7 @@ def link(object_files, destination_file=None, overwrite_destination_file=True, a *_tf_link_flags, *_include_flags, *additional_link_flags, + '-lcudart', _shared_object_flag, _output_flag] if not destination_file: -- GitLab