floraachy
/
OpenBLAS
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add nvfortran

This commit is contained in:
Martin Kroeker 2020-12-19 22:09:57 +01:00 committed by GitHub
parent b212a2fb9f
commit b859b6e79d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
f_check
View File

@ -32,7 +32,7 @@ if ($compiler eq "") {
"xlf95", "xlf90", "xlf", "xlf95", "xlf90", "xlf",
"ppuf77", "ppuf95", "ppuf90", "ppuxlf", "ppuf77", "ppuf95", "ppuf90", "ppuxlf",
"pathf90", "pathf95", "pathf90", "pathf95",
"pgf95", "pgf90", "pgf77", "pgf95", "pgf90", "pgf77", "pgfortran", "nvfortran",
"flang", "egfortran", "flang", "egfortran",
"ifort"); "ifort");
@ -64,7 +64,6 @@ if ($compiler eq "") {
if (!$?) { if (!$?) {
$data = `$compiler -O2 -S ftest.f > /dev/null 2>&1 && cat ftest.s && rm -f ftest.s`; $data = `$compiler -O2 -S ftest.f > /dev/null 2>&1 && cat ftest.s && rm -f ftest.s`;
if ($data =~ /zhoge_/) { if ($data =~ /zhoge_/) {
$bu = "_"; $bu = "_";
} }
@ -87,7 +86,7 @@ if ($compiler eq "") {
if ($compiler =~ /flang/) { if ($compiler =~ /flang/) {
$vendor = FLANG; $vendor = FLANG;
$openmp = "-fopenmp"; $openmp = "-fopenmp";
} elsif ($compiler =~ /pgf/) { } elsif ($compiler =~ /pgf/ || $compiler =~ /nvf/) {
$vendor = PGI; $vendor = PGI;
$openmp = "-mp"; $openmp = "-mp";
} else { } else {
@ -123,7 +122,7 @@ if ($compiler eq "") {
$openmp = "-mp"; $openmp = "-mp";
} }
if ($data =~ /PGF/) { if ($data =~ /PGF/ || $data =~ /NVF/) {
$vendor = PGI; $vendor = PGI;
$openmp = "-mp"; $openmp = "-mp";
} }
@ -177,7 +176,7 @@ if ($compiler eq "") {
$openmp = "-mp"; $openmp = "-mp";
} }
if ($compiler =~ /pgf/) { if ($compiler =~ /pgf/ || $compiler =~ /nvf/) {
$vendor = PGI; $vendor = PGI;
$bu = "_"; $bu = "_";
$openmp = "-mp"; $openmp = "-mp";
@ -330,7 +329,7 @@ if ($link ne "") {
$flags =~ s/\@/\,/g; $flags =~ s/\@/\,/g;
$linker_L .= "-Wl,". $flags . " " ; $linker_L .= "-Wl,". $flags . " " ;
} }
if ($flags =~ /-lgomp/ && $ENV{"CC"} =~ /clang/) { if ($flags =~ /-lgomp/ && $CC =~ /clang/) {
$flags = "-lomp"; $flags = "-lomp";
} }