Home
last modified time | relevance | path

Searched refs:deviceType (Results 1 – 5 of 5) sorted by relevance

/petsc/src/sys/objects/device/interface/
H A Ddevice.cxx665 const auto deviceType = PetscDeviceTypeCast(i); in PetscDeviceInitializeFromOptions_Internal() local
668 …PetscCall(PetscDeviceInitializeTypeFromOptions_Private(comm, deviceType, defaultDeviceId, defaultV… in PetscDeviceInitializeFromOptions_Internal()
669 if (PetscDeviceConfiguredFor_Internal(deviceType)) { in PetscDeviceInitializeFromOptions_Internal()
674 deviceContextInitDevice = deviceType; in PetscDeviceInitializeFromOptions_Internal()
675 …evice %s set as default device type due to eager initialization\n", PetscDeviceTypes[deviceType])); in PetscDeviceInitializeFromOptions_Internal()
678 …deviceType != PETSC_DEVICE_HOST) PetscCheck(!defaultDeviceIdSet || (deviceType != deviceContextIni… in PetscDeviceInitializeFromOptions_Internal()
H A Dpetscdevice_interface_internal.hpp179 …OptionItems PetscOptionsObject, std::pair<PetscDeviceType, PetscBool> &deviceType, std::pair<Petsc… in PetscDeviceContextQueryOptions_Internal() argument
181 auto dtype = static_cast<PetscInt>(deviceType.first); in PetscDeviceContextQueryOptions_Internal()
186 …Device", PetscDeviceTypes, PETSC_DEVICE_MAX, PetscDeviceTypes[dtype], &dtype, &deviceType.second)); in PetscDeviceContextQueryOptions_Internal()
188 deviceType.first = PetscDeviceTypeCast(dtype); in PetscDeviceContextQueryOptions_Internal()
/petsc/include/
H A Dpetsclandau.h182 LandauDeviceType deviceType; member
213 LandauDeviceType deviceType; member
/petsc/src/ts/utils/dmplexlandau/
H A Dplexland.c31 if (maps[0].deviceType != LANDAU_CPU) { in LandauGPUMapsDestroy()
33 …if (maps[0].deviceType == LANDAU_KOKKOS) PetscCall(LandauKokkosDestroyMatMaps(maps, maps[0].numgri… in LandauGPUMapsDestroy()
195 …PetscCheck(mtype == PETSC_MEMTYPE_HOST || ctx->deviceType != LANDAU_CPU, ctx->comm, PETSC_ERR_ARG_… in LandauFormJacobian_Internal()
202 if (ctx->deviceType == LANDAU_KOKKOS) { in LandauFormJacobian_Internal()
1144 ctx->deviceType = LANDAU_KOKKOS; in ProcessOptions()
1147 ctx->deviceType = LANDAU_CPU; in ProcessOptions()
1153 ctx->deviceType = LANDAU_CPU; in ProcessOptions()
1157 ctx->deviceType = LANDAU_KOKKOS; in ProcessOptions()
1512 maps[grid].deviceType = ctx->deviceType; in CreateStaticData()
1611 …if (ctx->deviceType == LANDAU_KOKKOS) PetscCall(LandauKokkosCreateMatMaps(maps, pointMaps, Nf, gri… in CreateStaticData()
[all …]
/petsc/src/ts/utils/dmplexlandau/kokkos/
H A Dlandau.kokkos.cxx107 h_maps.deviceType = maps[grid].deviceType; in LandauKokkosCreateMatMaps()