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