diff --git a/src/pystencils_autodiff/tensorflow_jit.py b/src/pystencils_autodiff/tensorflow_jit.py index e5312b02714bf39a20780be235c3b0d54dc03c15..916912a9b775b469ca948322b0ed74296e3e7d24 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: