Lines Matching refs:Kokkos

11 Kokkos::DefaultExecutionSpace *PetscKokkosExecutionSpacePtr = nullptr;
18   PetscCallCXX(Kokkos::kokkos_free(PetscScalarPool));  in PetscKokkosFinalize_Private()
21     PetscCallCXX(Kokkos::finalize());  in PetscKokkosFinalize_Private()
30   *isInitialized = Kokkos::is_initialized() ? PETSC_TRUE : PETSC_FALSE;  in PetscKokkosIsInitialized_Private()
38   if (!Kokkos::is_initialized()) {  in PetscKokkosInitializeCheck()
40     auto args = Kokkos::InitializationSettings();  in PetscKokkosInitializeCheck()
42     auto args = Kokkos::InitArguments{}; /* use default constructor */  in PetscKokkosInitializeCheck()
72     PetscCallCXX(Kokkos::initialize(args));  in PetscKokkosInitializeCheck()
85 …if (dtype == PETSC_DEVICE_CUDA) PetscCallCXX(PetscKokkosExecutionSpacePtr = new Kokkos::DefaultExe…  in PetscKokkosInitializeCheck()
87 …if (dtype == PETSC_DEVICE_HIP) PetscCallCXX(PetscKokkosExecutionSpacePtr = new Kokkos::DefaultExec…  in PetscKokkosInitializeCheck()
92 …PetscCallCXX(PetscKokkosExecutionSpacePtr = new Kokkos::DefaultExecutionSpace(*(sycl::queue *)hand…  in PetscKokkosInitializeCheck()
97     PetscCallCXX(PetscKokkosExecutionSpacePtr = new Kokkos::DefaultExecutionSpace());  in PetscKokkosInitializeCheck()
103 …PetscCallCXX(PetscScalarPool = static_cast<PetscScalar *>(Kokkos::kokkos_malloc(sizeof(PetscScalar…  in PetscKokkosInitializeCheck()