提交 f6853a87 编写于 作者: 李超

Merge branch 'fix_deconv_bug' into 'master'

fix: set Deconv2d filter's opencl buffer type to CONV2D_FILTER

See merge request applied-machine-learning/sysml/mace!1293
...@@ -248,7 +248,7 @@ void RegisterDeconv2D(OpRegistry *op_registry) { ...@@ -248,7 +248,7 @@ void RegisterDeconv2D(OpRegistry *op_registry) {
op_registry, op_registry,
OpConditionBuilder("Deconv2D").SetInputMemoryTypeSetter( OpConditionBuilder("Deconv2D").SetInputMemoryTypeSetter(
[](OpConditionContext *context) -> void { [](OpConditionContext *context) -> void {
SetFilterMemoryType(context, OpenCLBufferType::DW_CONV2D_FILTER); SetFilterMemoryType(context, OpenCLBufferType::CONV2D_FILTER);
if (context->device()->device_type() == DeviceType::GPU) { if (context->device()->device_type() == DeviceType::GPU) {
FrameworkType framework_type = FrameworkType framework_type =
static_cast<FrameworkType>( static_cast<FrameworkType>(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册