Skip to content

Commit 37a4ca7

Browse files
authored
Merge pull request #5025 from martin-frbg/nvidia_arm64
Add target-specific options to enable ARM64 SVE with the NVIDIA compiler
2 parents f2be482 + 1c4401e commit 37a4ca7

File tree

1 file changed

+27
-0
lines changed

1 file changed

+27
-0
lines changed

Makefile.arm64

Lines changed: 27 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -351,4 +351,31 @@ endif
351351

352352
endif
353353

354+
else
355+
# NVIDIA HPC options necessary to enable SVE in the compiler
356+
ifeq ($(CORE), THUNDERX2T99)
357+
CCOMMON_OPT += -tp=thunderx2t99
358+
FCOMMON_OPT += -tp=thunderx2t99
359+
endif
360+
ifeq ($(CORE), NEOVERSEN1)
361+
CCOMMON_OPT += -tp=neoverse-n1
362+
FCOMMON_OPT += -tp=neoverse-n1
363+
endif
364+
ifeq ($(CORE), NEOVERSEV1)
365+
CCOMMON_OPT += -tp=neoverse-v1
366+
FCOMMON_OPT += -tp=neoverse-v1
367+
endif
368+
ifeq ($(CORE), NEOVERSEV2)
369+
CCOMMON_OPT += -tp=neoverse-v2
370+
FCOMMON_OPT += -tp=neoverse-v2
371+
endif
372+
ifeq ($(CORE), ARMV8SVE)
373+
CCOMMON_OPT += -tp=neoverse-v2
374+
FCOMMON_OPT += -tp=neoverse-v2
375+
endif
376+
ifeq ($(CORE), ARMV9SVE)
377+
CCOMMON_OPT += -tp=neoverse-v2
378+
FCOMMON_OPT += -tp=neoverse-v2
379+
endif
380+
354381
endif

0 commit comments

Comments
 (0)