From 60c9f4717a9c759ccd7254a3d424ed8b0863b10b Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Fri, 29 Nov 2019 18:14:58 +0100
Subject: [PATCH] Put openmp in spot where its allowed
---
src/pystencils_autodiff/tensorflow_jit.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/pystencils_autodiff/tensorflow_jit.py b/src/pystencils_autodiff/tensorflow_jit.py
index 4bd7923..2747d6b 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:
--
GitLab