Home
last modified time | relevance | path

Searched refs:device_type (Results 1 – 9 of 9) sorted by relevance

/petsc/src/ksp/ksp/utils/lmvm/blas_cyclic/cupm/
H A Dblas_cyclic_cupm.cxx33 PetscDeviceType device_type; in AXPBYCyclic_CUPM_Private() local
37 PetscCall(PetscDeviceContextGetDeviceType(dctx, &device_type)); in AXPBYCyclic_CUPM_Private()
38 switch (device_type) { in AXPBYCyclic_CUPM_Private()
50 …TERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "Unsupported device type %s", PetscDeviceTypes[device_type]); in AXPBYCyclic_CUPM_Private()
58 PetscDeviceType device_type; in DMVCyclic_CUPM_Private() local
62 PetscCall(PetscDeviceContextGetDeviceType(dctx, &device_type)); in DMVCyclic_CUPM_Private()
63 switch (device_type) { in DMVCyclic_CUPM_Private()
75 …TERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "Unsupported device type %s", PetscDeviceTypes[device_type]); in DMVCyclic_CUPM_Private()
83 PetscDeviceType device_type; in DSVCyclic_CUPM_Private() local
87 PetscCall(PetscDeviceContextGetDeviceType(dctx, &device_type)); in DSVCyclic_CUPM_Private()
[all …]
/petsc/src/ksp/ksp/utils/lmvm/dense/cd_cupm/
H A Dcd_cupm.cxx104 PetscDeviceType device_type; in MatUpperTriangularSolveInPlace_CUPM() local
108 PetscCall(PetscDeviceContextGetDeviceType(dctx, &device_type)); in MatUpperTriangularSolveInPlace_CUPM()
109 switch (device_type) { in MatUpperTriangularSolveInPlace_CUPM()
121 …TERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "Unsupported device type %s", PetscDeviceTypes[device_type]); in MatUpperTriangularSolveInPlace_CUPM()
131 PetscDeviceType device_type; in MatUpperTriangularSolveInPlaceCyclic_CUPM() local
135 PetscCall(PetscDeviceContextGetDeviceType(dctx, &device_type)); in MatUpperTriangularSolveInPlaceCyclic_CUPM()
136 switch (device_type) { in MatUpperTriangularSolveInPlaceCyclic_CUPM()
148 …TERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "Unsupported device type %s", PetscDeviceTypes[device_type]); in MatUpperTriangularSolveInPlaceCyclic_CUPM()
/petsc/src/sys/objects/device/impls/sycl/
H A Dsycldevice.sycl.cxx97 return ::sycl::device::get_devices(::sycl::info::device_type::gpu)[id]; in chooseSYCLDevice_()
150 …std::vector<::sycl::device> gpu_devices = ::sycl::device::get_devices(::sycl::info::device_type::g… in initialize()
H A Dsyclcontext.sycl.cxx79 …::device(::sycl::cpu_selector_v) : ::sycl::device::get_devices(::sycl::info::device_type::gpu)[id]; in setUp()
/petsc/src/binding/petsc4py/src/petsc4py/PETSc/
H A Ddlpack.pxi39 PetscDLDeviceType device_type
H A DDevice.pyx196 def setDefaultType(device_type: Type | str) -> None:
206 cdef PetscDeviceType cdevice_type = asDeviceType(device_type)
H A DVec.pyx610 cdef PetscDLDeviceType dltype = ptr.dl_tensor.ctx.device_type
638 …self.set_attr('__dltensor_ctx__', (ptr.dl_tensor.ctx.device_type, ptr.dl_tensor.ctx.device_id, ndi…
702 …self.set_attr('__dltensor_ctx__', (ptr.dl_tensor.ctx.device_type, ptr.dl_tensor.ctx.device_id, ndi…
756 (device_type, device_id, ndim, shape, strides) = vec_get_dlpack_ctx(self)
757 hostmem = (device_type == kDLCPU)
793 ctx.device_type = device_type
H A DMat.pyx5999 (device_type, device_id, ndim, shape, strides) = mat_get_dlpack_ctx(self)
6000 hostmem = (device_type == kDLCPU)
6036 ctx.device_type = device_type
/petsc/src/vec/vec/impls/seq/kokkos/
H A Dveckok.kokkos.cxx21 using ExecutionSpace = typename PetscScalarKokkosViewType<MemorySpace>::traits::device_type; in VecGetKokkosView_Private()
36 using ExecutionSpace = typename PetscScalarKokkosViewType<MemorySpace>::traits::device_type; in VecRestoreKokkosView_Private()
50 using ExecutionSpace = typename PetscScalarKokkosViewType<MemorySpace>::traits::device_type; in VecGetKokkosView()