MCPcopy Create free account
hub / github.com/MegEngine/MegEngine / megcoreGetROCMStream

Function megcoreGetROCMStream

dnn/include/megcore_rocm.h:52–58  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

50}
51
52static inline megcoreStatus_t megcoreGetROCMStream(
53 megcoreComputingHandle_t handle, hipStream_t* stream) {
54 megcore::ROCMContext ctx;
55 auto ret = megcore::getROCMContext(handle, &ctx);
56 *stream = ctx.stream;
57 return ret;
58}
59
60#include "megdnn/internal/visibility_epilogue.h"
61

Callers 2

BenchmarkEnvMethod · 0.85
TEST_FFunction · 0.85

Calls

no outgoing calls

Tested by 2

BenchmarkEnvMethod · 0.68
TEST_FFunction · 0.68