xref: /petsc/config/examples/arch-nvhpc.py (revision 2f613bf53f46f9356e00a2ca2bd69453be72fc31)
1#!/usr/bin/python3
2if __name__ == '__main__':
3  import sys
4  import os
5  sys.path.insert(0, os.path.abspath('config'))
6  import configure
7  configure_options = [
8    '--download-kokkos',
9    '--download-kokkos-kernels',
10    '--with-cuda=1',
11    '--with-cxx-dialect=c++14',
12    '--with-debugging=0',
13    # Need to use g++ as host compiler for NVCC (tested with 7.5.0) to compile kokkos lambdas
14    'CUDAFLAGS=-ccbin g++',
15    # Uses NVC (PGI) compilers for MPI wrappers
16    'CFLAGS=-g -nomp -tp p7-64',
17    'CXXFLAGS=-g -nomp -tp p7-64',
18    'FFLAGS=-g -nomp -tp p7-64',
19    'PETSC_ARCH=arch-nvhpc',
20  ]
21  configure.petsc_configure(configure_options)
22