| 36 | } |
| 37 | |
| 38 | static inline megcoreStatus_t megcoreGetCUDAStream( |
| 39 | megcoreComputingHandle_t handle, cudaStream_t* stream) { |
| 40 | megcore::CudaContext ctx; |
| 41 | auto ret = megcore::getCUDAContext(handle, &ctx); |
| 42 | *stream = ctx.stream; |
| 43 | return ret; |
| 44 | } |
| 45 | |
| 46 | #include "megdnn/internal/visibility_epilogue.h" |
| 47 |
no outgoing calls