Link cudart

3a3bd16a · Stephan Seitz · 4b8386c3 · 3a3bd16a
Commit 3a3bd16a authored 5 years ago by Stephan Seitz
--- 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: