Skip to content
Snippets Groups Projects
Commit accb0442 authored by Stephan Seitz's avatar Stephan Seitz
Browse files

Avoid writing to config file tensorflow_jit

parent 40bec040
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
src/pystencils_autodiff/tensorflow_jit.py
+ 0
− 2
View file @ accb0442
...@@ -8,7 +8,6 @@ ...@@ -8,7 +8,6 @@
""" """
import hashlib import hashlib
import json
import subprocess import subprocess
import sysconfig import sysconfig
from os.path import exists, join from os.path import exists, join
...@@ -106,7 +105,6 @@ if pystencils.gpucuda.cudajit.USE_FAST_MATH: ...@@ -106,7 +105,6 @@ if pystencils.gpucuda.cudajit.USE_FAST_MATH:
def compile_file(file, use_nvcc=False, nvcc='nvcc', overwrite_destination_file=True, additional_compile_flags=[]): def compile_file(file, use_nvcc=False, nvcc='nvcc', overwrite_destination_file=True, additional_compile_flags=[]):
if 'tensorflow_host_compiler' not in get_compiler_config(): if 'tensorflow_host_compiler' not in get_compiler_config():
get_compiler_config()['tensorflow_host_compiler'] = get_compiler_config()['command'] get_compiler_config()['tensorflow_host_compiler'] = get_compiler_config()['command']
write_file(pystencils.cpu.cpujit.get_configuration_file_path(), json.dumps(pystencils.cpu.cpujit._config))
destination_file = file + '.o' destination_file = file + '.o'
if use_nvcc: if use_nvcc:
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment