提交 c62ddba2 编写于 作者: M Megvii Engine Team

feat(dnn/opencl): optimize heuristic rule

GitOrigin-RevId: 971c93d926c774f574d75a1ee8e1e37a4295d9c0
上级 6e839407
...@@ -261,7 +261,9 @@ void Image2DPackedTensorFormatBase<PIXEL_SIZE>::assert_valid( ...@@ -261,7 +261,9 @@ void Image2DPackedTensorFormatBase<PIXEL_SIZE>::assert_valid(
layout.shape[layout.ndim - 1]); layout.shape[layout.ndim - 1]);
megdnn_assert( megdnn_assert(
layout.dtype.valid() && !layout.dtype.is_quantized_lowbit() && layout.dtype.valid() && !layout.dtype.is_quantized_lowbit() &&
layout.ndim > m_align_axis); layout.ndim > m_align_axis,
"dtype=%s ndim=%zu align=%zu is_quantized_lowbit=%d", layout.dtype.name(),
layout.ndim, m_align_axis, layout.dtype.is_quantized_lowbit());
ptrdiff_t first_non_zero_stride = 0; ptrdiff_t first_non_zero_stride = 0;
for (int i = layout.ndim - 1; i >= 0; --i) { for (int i = layout.ndim - 1; i >= 0; --i) {
megdnn_assert(layout.shape[i]); megdnn_assert(layout.shape[i]);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册