Skip to content
GitLab
Menu
Projects
Groups
Snippets
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
pycodegen
pystencils_autodiff
Commits
22007af0
Commit
22007af0
authored
Dec 17, 2019
by
Stephan Seitz
Browse files
Do not link against cudart as default
parent
7a292ebb
Changes
1
Hide whitespace changes
Inline
Side-by-side
src/pystencils_autodiff/tensorflow_jit.py
View file @
22007af0
...
...
@@ -62,7 +62,7 @@ def link(object_files,
destination_file
=
None
,
overwrite_destination_file
=
True
,
additional_link_flags
=
[],
link_cudart
=
Tru
e
):
link_cudart
=
Fals
e
):
"""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
=
Tru
e
):
link_cudart
=
Fals
e
):
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
:
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment