Actual source code: petscdevice_cupm.h

  1: #ifndef PETSCDEVICE_CUPM_H
  2: #define PETSCDEVICE_CUPM_H

  4: #include <petscmacros.h>
  5: #include <petscdevice_cuda.h>
  6: #include <petscdevice_hip.h>

  8: #if PetscDefined(USING_HCC) && PetscDefined(USING_NVCC)
  9:   #error using both nvcc and hipcc at the same time?
 10: #endif

 12: #if PetscDefined(HAVE_CUDA) || PetscDefined(HAVE_HIP)
 13:   #define PETSC_HAVE_CUPM 1
 14: #endif

 16: #if PetscDefined(USING_NVCC) || PetscDefined(USING_HCC)
 17:   #define PETSC_USING_CUPMCC 1
 18: #endif

 20: #endif // PETSCDEVICE_CUPM_H