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

Method stream

dnn/src/rocm/handle.h:45–45  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

43 int device_id() const { return m_device_id; }
44
45 hipStream_t stream() const { return megcore_context().stream; }
46 miopenHandle_t miopen_handle() { return m_miopen_handle; }
47 rocblas_handle get_rocblas_handle() { return m_rocblas_handle; }
48 dt_float32* zero_device() { return &m_const_scalars->f32[0]; }

Callers 1

hip_streamFunction · 0.45

Calls

no outgoing calls

Tested by

no test coverage detected