Lines Matching defs:device

79   CUdevice device;
80   CUDA_REPORT_IF_ERROR(cuDeviceGet(&device, /*ordinal=*/defaultDevice));
81   return device;
84 // Make the primary context of the current default device current for the
90     // Static reference to CUDA primary context for device ordinal
175     CUdevice device = getDefaultCuDevice();
176     CUDA_REPORT_IF_ERROR(cuDeviceGet(&device, /*ordinal=*/defaultDevice));
179         device));
183               "shared memory (%dkb) for this device\n",
335 extern "C" MLIR_CUDA_WRAPPERS_EXPORT void mgpuSetDefaultDevice(int32_t device) {
336   defaultDevice = device;
354     CUdevice device = getDefaultCuDevice();
355     CUDA_REPORT_IF_ERROR(cuDeviceGet(&device, /*ordinal=*/defaultDevice));
358         device));
362               "shared memory (%dkb) for this device\n",
510   // Copy created tensor map to device