Home
last modified time | relevance | path

Searched refs:cudaGetDeviceCount (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/tests/ut/cpp/stub/runtime/
Dcuda_runtime_api.cc32 cudaError_t cudaGetDeviceCount(int *count) { return cudaSuccess; } in cudaGetDeviceCount() function
Dcuda_runtime_api.h44 cudaError_t cudaGetDeviceCount(int *count);
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dcuda_driver.cc235 auto ret = cudaGetDeviceCount(&dev_count); in device_count()
/third_party/mindspore/mindspore/lite/src/delegate/tensorrt/
Dtensorrt_utils.cc273 ret = cudaGetDeviceCount(&deviceCnt); in SetCudaDevice()
Dtensorrt_subgraph.cc117 cudaError ret = cudaGetDeviceCount(&deviceCnt); in SupportFP16()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/
Ddevice_queue_op.cc48 cudaGetDeviceCount(&device_count); in DeviceQueueOp()