Added support for NEC SDK

This commit is contained in:
Patrick Lipka 2022-09-29 21:21:03 +02:00
parent 05af2edff9
commit 33bb44d54b
1 changed files with 28 additions and 1 deletions

View File

@ -1,4 +1,5 @@
import os import os
import sys
from lib.shell import get_from_command from lib.shell import get_from_command
@ -42,6 +43,13 @@ def set_toolchain(compiler, mpi):
cc = "mpicc" cc = "mpicc"
cxx = "mpicxx" cxx = "mpicxx"
fc = "mpif90" fc = "mpif90"
elif mpi == "necmpi":
cc = "mpicc -vh"
cxx = "mpic++ -vh"
fc = "mpinfort -vh"
os.environ["NMPI_CC_H"] = "gcc"
os.environ["NMPI_CXX_H"] = "g++"
os.environ["NMPI_FC_H"] = "gfortran"
elif compiler == "aocc": elif compiler == "aocc":
if mpi == "hpcx" or mpi == "openmpi": if mpi == "hpcx" or mpi == "openmpi":
cc = "mpicc" cc = "mpicc"
@ -56,6 +64,13 @@ def set_toolchain(compiler, mpi):
cc = "\"mpicc -cc=clang\"" cc = "\"mpicc -cc=clang\""
cxx = "\"mpicxx -cxx=clang++\"" cxx = "\"mpicxx -cxx=clang++\""
fc = "\"mpif90 -fc=flang\"" fc = "\"mpif90 -fc=flang\""
elif mpi == "necmpi":
cc = "mpicc -vh"
cxx = "mpic++ -vh"
fc = "mpinfort -vh"
os.environ["NMPI_CC_H"] = "clang"
os.environ["NMPI_CXX_H"] = "clang++"
os.environ["NMPI_FC_H"] = "flang"
elif compiler == "intel": elif compiler == "intel":
if mpi == "hpcx" or mpi == "openmpi": if mpi == "hpcx" or mpi == "openmpi":
cc = "mpicc" cc = "mpicc"
@ -70,6 +85,18 @@ def set_toolchain(compiler, mpi):
cc = "mpiicc" cc = "mpiicc"
cxx = "mpiicpc" cxx = "mpiicpc"
fc = "mpiifort" fc = "mpiifort"
elif mpi == "necmpi":
cc = "mpicc -vh"
cxx = "mpic++ -vh"
fc = "mpinfort -vh"
os.environ["NMPI_CC_H"] = "icc"
os.environ["NMPI_CXX_H"] = "icpc"
os.environ["NMPI_FC_H"] = "ifort"
elif compiler == "nec":
# only NEC MPI
cc = "mpincc"
cxx = "mpinc++"
fc = "mpinfort"
# set environment variables # set environment variables
os.environ["CC"] = cc os.environ["CC"] = cc