xref: /petsc/config/examples/arch-nvhpc.py (revision ac8a123e01879214fde757ebe5fc71983e4a156c)
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-debugging=0',
12*04f151b2SStefano Zampini    # Need to use g++ as host compiler for NVCC (tested with 7.5.0) to compile kokkos lambdas
13*04f151b2SStefano Zampini    'CUDAFLAGS=-ccbin g++',
14*04f151b2SStefano Zampini    # Uses NVC (PGI) compilers for MPI wrappers
15*04f151b2SStefano Zampini    'CFLAGS=-g -nomp -tp p7-64',
16*04f151b2SStefano Zampini    'CXXFLAGS=-g -nomp -tp p7-64',
17*04f151b2SStefano Zampini    'FFLAGS=-g -nomp -tp p7-64',
18*04f151b2SStefano Zampini    'PETSC_ARCH=arch-nvhpc',
19*04f151b2SStefano Zampini  ]
20*04f151b2SStefano Zampini  configure.petsc_configure(configure_options)
21