ERF
Energy Research and Forecasting: An Atmospheric Modeling Code
ERF_Kokkos.H
Go to the documentation of this file.
1 #ifndef ERF_KOKKOS_H
2 #define ERF_KOKKOS_H
3 using KokkosHostDevice = Kokkos::Device<Kokkos::Serial, Kokkos::HostSpace>;
4 #if defined(ERF_USE_CUDA)
5 using KokkosDefaultDevice = Kokkos::Device<Kokkos::Cuda, Kokkos::CudaSpace>;
6 #elif defined(ERF_USE_HIP)
7 using KokkosDefaultDevice = Kokkos::Device<Kokkos::HIP, Kokkos::HIPSpace>;
8 #elif defined(ERF_USE_SYCL)
9 using KokkosDefaultDevice = Kokkos::Device<Kokkos::Experimental::SYCL, Kokkos::Experimental::SYCLDeviceUSMSpace>;
10 #else
12 #endif
13 using layout_t = Kokkos::LayoutRight;
15 using realHost1d_k = Kokkos::View<RealT* , KokkosHostDevice>;
16 using realHost2d_k = Kokkos::View<RealT**, layout_t, KokkosHostDevice>;
17 using real1d_k = Kokkos::View<RealT* , KokkosDefaultDevice>;
18 using real2d_k = Kokkos::View<RealT** , layout_t, KokkosDefaultDevice>;
19 using real3d_k = Kokkos::View<RealT***, layout_t, KokkosDefaultDevice>;
20 using int1d_k = Kokkos::View<int* , KokkosDefaultDevice>;
21 using int2d_k = Kokkos::View<int** , layout_t, KokkosDefaultDevice>;
22 using int3d_k = Kokkos::View<int***, layout_t, KokkosDefaultDevice>;
23 #endif
KokkosHostDevice KokkosDefaultDevice
Definition: ERF_Kokkos.H:11
Kokkos::View< RealT *, KokkosDefaultDevice > real1d_k
Definition: ERF_Kokkos.H:17
Kokkos::View< int *, KokkosDefaultDevice > int1d_k
Definition: ERF_Kokkos.H:20
amrex::Real RealT
Definition: ERF_Kokkos.H:14
Kokkos::View< RealT ***, layout_t, KokkosDefaultDevice > real3d_k
Definition: ERF_Kokkos.H:19
Kokkos::View< RealT **, layout_t, KokkosDefaultDevice > real2d_k
Definition: ERF_Kokkos.H:18
Kokkos::View< RealT **, layout_t, KokkosHostDevice > realHost2d_k
Definition: ERF_Kokkos.H:16
Kokkos::View< int **, layout_t, KokkosDefaultDevice > int2d_k
Definition: ERF_Kokkos.H:21
Kokkos::Device< Kokkos::Serial, Kokkos::HostSpace > KokkosHostDevice
Definition: ERF_Kokkos.H:3
Kokkos::LayoutRight layout_t
Definition: ERF_Kokkos.H:13
Kokkos::View< int ***, layout_t, KokkosDefaultDevice > int3d_k
Definition: ERF_Kokkos.H:22
Kokkos::View< RealT *, KokkosHostDevice > realHost1d_k
Definition: ERF_Kokkos.H:15
amrex::Real Real
Definition: ERF_ShocInterface.H:16