Added support for NEC SDK
This commit is contained in:
		
							parent
							
								
									05af2edff9
								
							
						
					
					
						commit
						33bb44d54b
					
				|  | @ -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 | ||||||
|  |  | ||||||
		Loading…
	
		Reference in New Issue