Skip to content

Commit ae21250

Browse files
committed
Try to improve NVPL detection
Signed-off-by: Jan André Reuter <[email protected]>
1 parent 00c7fb4 commit ae21250

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

easybuild/easyblocks/f/flexiblas.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -125,7 +125,7 @@ def configure_step(self):
125125
if blas_lib == 'imkl':
126126
# For MKL there is gf_lp64 vs. intel_lp64 and gnu_thread vs. intel_thread (vs. sequential)
127127
# For gf_lp64 vs intel_lp64 the difference is in the ABI for [sz]dot[uc], which FlexiBLAS
128-
# can transparenly wrap.
128+
# can transparently wrap.
129129
# gnu_thread vs intel_thread links to libgomp vs. libiomp5 for the OpenMP library.
130130
mkl_gnu_libs = "mkl_gf_lp64;mkl_gnu_thread;mkl_core;gomp;pthread;m;dl"
131131
mkl_intel_libs = "mkl_intel_lp64;mkl_intel_thread;mkl_core;iomp5;pthread;m;dl"
@@ -140,6 +140,11 @@ def configure_step(self):
140140
configopts[key] = mkl_compiler_mapping[comp_family]
141141
except KeyError:
142142
raise EasyBuildError("Compiler family not supported yet: %s", comp_family)
143+
elif blas_lib == "NVPL":
144+
# NVPL libraries do not explicitly link any OpenMP runtime,
145+
# but try to lazily and dynamically find the OpenMP runtime in the
146+
# built program. Supported are libgomp, libomp and libnvomp.
147+
configopts[key] = "nvpl_blas_lp64_gomp;nvpl_blas_core"
143148
elif blas_lib == 'AOCL_mt':
144149
configopts[key] = 'blis-mt'
145150
else:

0 commit comments

Comments
 (0)