Home
last modified time | relevance | path

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

/third_party/mindspore/tests/ut/cpp/stub/runtime/
Dcuda_runtime_api.cc34 cudaError_t cudaSetDevice(int device) { return cudaSuccess; } in cudaSetDevice() function
Dcuda_runtime_api.h45 cudaError_t cudaSetDevice(int device);
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dcuda_driver.cc243 auto ret = cudaSetDevice(index); in SetDevice()
Dgpu_buffer_mgr.cc107 auto ret = cudaSetDevice(cur_dev_id_); in set_device()
/third_party/mindspore/mindspore/lite/src/delegate/tensorrt/
Dtensorrt_utils.cc282 ret = cudaSetDevice(set_device_id); in SetCudaDevice()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/
Ddevice_queue_op.cc369 auto ret = cudaSetDevice(rank_id_); in SetThreadDevice()