diff --git a/src/pystencils_autodiff/tensorflow_jit.py b/src/pystencils_autodiff/tensorflow_jit.py index 4bd7923efd4dd1989857b6cd052ca6096fff5d02..2747d6b25d3222439802d1a2da35232560da9871 100644 --- a/src/pystencils_autodiff/tensorflow_jit.py +++ b/src/pystencils_autodiff/tensorflow_jit.py @@ -176,7 +176,7 @@ def compile_file(file, _output_flag] if openmp: - command_prefix.append(_output_flag) + command_prefix.insert(2, _output_flag) destination_file = f'{file}_{_hash(".".join(command_prefix).encode()).hexdigest()}{_object_file_extension}' if not exists(destination_file) or overwrite_destination_file: