From 9218e0f29ed194530fcd4b5e9bdcce82397674c1 Mon Sep 17 00:00:00 2001 From: mloubout Date: Tue, 18 Feb 2025 16:17:41 -0500 Subject: [PATCH] arch: add missing openmp flag for aomp compiler --- devito/arch/compiler.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/devito/arch/compiler.py b/devito/arch/compiler.py index e1270095e5..aa8cfa0795 100644 --- a/devito/arch/compiler.py +++ b/devito/arch/compiler.py @@ -549,6 +549,9 @@ def __init_finalize__(self, **kwargs): if not configuration['safe-math']: self.cflags.append('-ffast-math') + if language == 'openmp': + self.ldflags += ['-fopenmp'] + if isinstance(platform, NvidiaDevice): self.cflags.remove('-std=c99') elif platform is AMDGPUX: @@ -556,7 +559,6 @@ def __init_finalize__(self, **kwargs): # Add flags for OpenMP offloading if language in ['C', 'openmp']: self.ldflags += ['-target', 'x86_64-pc-linux-gnu'] - self.ldflags += ['-fopenmp'] self.ldflags += [f'--offload-arch={platform.march}'] elif platform in [POWER8, POWER9]: # It doesn't make much sense to use AOMP on Power, but it should work