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

Method one_device

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

Source from the content-addressed store, hash-verified

47 rocblas_handle get_rocblas_handle() { return m_rocblas_handle; }
48 dt_float32* zero_device() { return &m_const_scalars->f32[0]; }
49 dt_float32* one_device() { return &m_const_scalars->f32[1]; }
50#if !MEGDNN_DISABLE_FLOAT16
51 __half* zero_device_h() { return &m_const_scalars->f16[0].hip_x; }
52 __half* one_device_h() { return &m_const_scalars->f16[1].hip_x; }

Callers 2

execMethod · 0.45
execMethod · 0.45

Calls

no outgoing calls

Tested by

no test coverage detected