56 lines
1.5 KiB
Python
56 lines
1.5 KiB
Python
import sys
|
|
|
|
from numpy.distutils.fcompiler import FCompiler
|
|
|
|
compilers = ['NVHPCFCompiler']
|
|
|
|
class NVHPCFCompiler(FCompiler):
|
|
""" NVIDIA High Performance Computing (HPC) SDK Fortran Compiler
|
|
|
|
https://developer.nvidia.com/hpc-sdk
|
|
|
|
Since august 2020 the NVIDIA HPC SDK includes the compilers formerly known as The Portland Group compilers,
|
|
https://www.pgroup.com/index.htm.
|
|
See also `numpy.distutils.fcompiler.pg`.
|
|
"""
|
|
|
|
compiler_type = 'nv'
|
|
description = 'NVIDIA HPC SDK'
|
|
version_pattern = r'\s*(nvfortran|(pg(f77|f90|fortran)) \(aka nvfortran\)) (?P<version>[\d.-]+).*'
|
|
|
|
executables = {
|
|
'version_cmd': ["<F90>", "-V"],
|
|
'compiler_f77': ["nvfortran"],
|
|
'compiler_fix': ["nvfortran", "-Mfixed"],
|
|
'compiler_f90': ["nvfortran"],
|
|
'linker_so': ["<F90>"],
|
|
'archiver': ["ar", "-cr"],
|
|
'ranlib': ["ranlib"]
|
|
}
|
|
pic_flags = ['-fpic']
|
|
|
|
module_dir_switch = '-module '
|
|
module_include_switch = '-I'
|
|
|
|
def get_flags(self):
|
|
opt = ['-Minform=inform', '-Mnosecond_underscore']
|
|
return self.pic_flags + opt
|
|
|
|
def get_flags_opt(self):
|
|
return ['-fast']
|
|
|
|
def get_flags_debug(self):
|
|
return ['-g']
|
|
|
|
def get_flags_linker_so(self):
|
|
return ["-shared", '-fpic']
|
|
|
|
def runtime_library_dir_option(self, dir):
|
|
return '-R%s' % dir
|
|
|
|
if __name__ == '__main__':
|
|
from distutils import log
|
|
log.set_verbosity(2)
|
|
from numpy.distutils import customized_fcompiler
|
|
print(customized_fcompiler(compiler='nv').get_version())
|