Home
last modified time | relevance | path

Searched refs:device_id (Results 1 – 11 of 11) sorted by relevance

/libCEED/backends/magma/
H A Dceed-magma-common.c21 const int device_id = (device_spec) ? atoi(device_spec + 11) : -1; in CeedInit_Magma_common() local
27 if (device_id >= 0 && current_device_id != device_id) { in CeedInit_Magma_common()
28 magma_setdevice(device_id); in CeedInit_Magma_common()
29 current_device_id = device_id; in CeedInit_Magma_common()
33 data->device_id = current_device_id; in CeedInit_Magma_common()
35 magma_queue_create_from_hip(data->device_id, NULL, NULL, NULL, &(data->queue)); in CeedInit_Magma_common()
37 magma_queue_create_from_cuda(data->device_id, NULL, NULL, NULL, &(data->queue)); in CeedInit_Magma_common()
H A Dceed-magma-common.h14 magma_device_t device_id; member
/libCEED/backends/cuda/
H A Dceed-cuda-common.c22 const int device_id = (device_spec) ? atoi(device_spec + 11) : -1; in CeedInit_Cuda() local
26 if (device_id >= 0 && current_device_id != device_id) { in CeedInit_Cuda()
27 CeedCallCuda(ceed, cudaSetDevice(device_id)); in CeedInit_Cuda()
28 current_device_id = device_id; in CeedInit_Cuda()
32 data->device_id = current_device_id; in CeedInit_Cuda()
H A Dceed-cuda-common.h68 int device_id; member
H A Dceed-cuda-compile.cpp113 CeedCallCuda(ceed, cudaGetDeviceProperties(&prop, ceed_data->device_id)); in CeedCompileCore_Cuda()
/libCEED/backends/hip/
H A Dceed-hip-common.c21 const int device_id = (device_spec) ? atoi(device_spec + 11) : -1; in CeedInit_Hip() local
26 if (device_id >= 0 && current_device_id != device_id) { in CeedInit_Hip()
27 CeedCallHip(ceed, hipSetDevice(device_id)); in CeedInit_Hip()
28 current_device_id = device_id; in CeedInit_Hip()
32 data->device_id = current_device_id; in CeedInit_Hip()
H A Dceed-hip-common.h71 int device_id; member
H A Dceed-hip-compile.cpp83 CeedCallHip(ceed, hipGetDeviceProperties(&prop, ceed_data->device_id)); in CeedCompileCore_Hip()
/libCEED/backends/sycl/
H A Dceed-sycl-common.sycl.cpp21 const int device_id = (device_spec) ? atoi(device_spec + 11) : 0; in CeedInit_Sycl() local
40 if (device_count < device_id + 1) { in CeedInit_Sycl()
44 sycl::device sycl_device{sycl_devices[device_id]}; in CeedInit_Sycl()
/libCEED/
H A DREADME.md221 Users can specify a device for all CUDA, HIP, and MAGMA backends through adding `:device_id=#` afte…
224 > - `/gpu/cuda/gen:device_id=1`
/libCEED/backends/cuda-gen/
H A Dceed-cuda-gen-operator-build.cpp1262 CeedCallBackend(cudaGetDeviceProperties(&prop, ceed_data->device_id)); in CeedOperatorBuildKernel_Cuda_gen()
1678 CeedCallBackend(cudaGetDeviceProperties(&prop, ceed_data->device_id)); in CeedOperatorBuildKernelAssemblyAtPoints_Cuda_gen()
2202 CeedCallBackend(cudaGetDeviceProperties(&prop, ceed_data->device_id)); in CeedOperatorBuildKernelLinearAssembleQFunction_Cuda_gen()