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

fix(dnn/opencl): fix some bug for dnn opencl conv bias and relayout format

GitOrigin-RevId: b5bb07d90d316f2162c2441a017c8a01675921f5
上级 ca0e616f
......@@ -474,6 +474,8 @@ void RelayoutFormatImpl::exec(
src.layout.dtype.enumv() == DTypeEnum::Int32 ||
(src.layout.dtype.enumv() == DTypeEnum::Uint8 &&
dst.layout.dtype.enumv() == DTypeEnum::QuantizedS8) ||
(src.layout.dtype.enumv() == DTypeEnum::Uint8 &&
dst.layout.dtype.enumv() == DTypeEnum::Uint8) ||
src.layout.dtype.category() == DTypeCategory::QUANTIZED);
check_exec(src.layout, dst.layout, workspace.size);
HandleImpl* m_handle = static_cast<HandleImpl*>(handle());
......@@ -516,6 +518,7 @@ void RelayoutFormatImpl::exec(
DNN_INC_FLOAT16(cb(Float16, dt_float16));
cb(Quantized8Asymm, dt_uint8);
cb(QuantizedS8, dt_int8);
cb(Uint8, dt_uint8);
#undef cb
default:
megdnn_assert(0);
......
此差异由.gitattributes 抑制。
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册