Home
last modified time | relevance | path

Searched refs:defaultDevice (Results 1 – 2 of 2) sorted by relevance

/llvm-project-15.0.7/mlir/lib/ExecutionEngine/
H A DCudaRuntimeWrappers.cpp38 thread_local static int32_t defaultDevice = 0; variable
51 CUDA_REPORT_IF_ERROR(cuDeviceGet(&device, /*ordinal=*/defaultDevice)); in ScopedContext()
196 defaultDevice = device; in mgpuSetDefaultDevice()
H A DRocmRuntimeWrappers.cpp33 thread_local static int32_t defaultDevice = 0; variable
44 HIP_REPORT_IF_ERROR(hipDeviceGet(&device, /*ordinal=*/defaultDevice)); in ScopedContext()
206 defaultDevice = device; in mgpuSetDefaultDevice()