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