Home
last modified time | relevance | path

Searched refs:CUstream_st (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/tests/ut/cpp/stub/runtime/
Dcuda.h63 struct CUstream_st { struct
70 typedef struct CUstream_st *CUstream; argument
Dcuda_runtime_api.h31 struct CUstream_st { struct
/third_party/mindspore/mindspore/ccsrc/ps/ps_cache/gpu/
Dgpu_ps_cache.cc35 …CHECK_CUDA_RET_WITH_RETURN_ERROR_NOTRACE(cudaStreamCreate(reinterpret_cast<CUstream_st **>(&stream… in InitDevice()
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dcuda_driver.cc148 …auto ret = cudaStreamCreateWithFlags(reinterpret_cast<CUstream_st **>(stream), cudaStreamNonBlocki… in CreateStream()