未验证 提交 74074a8d 编写于 作者: Z zhupengyang 提交者: GitHub

fix xpu max size (#53505)

上级 eda0c588
...@@ -158,7 +158,7 @@ void Conv2dXPUInferMeta(const MetaTensor& x, ...@@ -158,7 +158,7 @@ void Conv2dXPUInferMeta(const MetaTensor& x,
} }
// set output and output max dims // set output and output max dims
out->set_dims(DDim(out_shape.data(), out_shape.size())); out->set_dims(DDim(out_shape.data(), out_shape.size()));
out_max->set_dims(phi::make_ddim({4})); out_max->set_dims(phi::make_ddim({6}));
} }
void EmbeddingWithEltwiseAddXPUInferMeta( void EmbeddingWithEltwiseAddXPUInferMeta(
...@@ -204,7 +204,7 @@ void FcXPUInferMeta(const MetaTensor& x, ...@@ -204,7 +204,7 @@ void FcXPUInferMeta(const MetaTensor& x,
out->set_dims(DDim(out_shape.data(), out_shape.size())); out->set_dims(DDim(out_shape.data(), out_shape.size()));
out->set_dtype(x.dtype()); out->set_dtype(x.dtype());
out->set_layout(x.layout()); out->set_layout(x.layout());
out_max->set_dims(w_max.dims()); out_max->set_dims(phi::make_ddim({6}));
out_max->set_dtype(x.dtype()); out_max->set_dtype(x.dtype());
out_max->set_layout(x.layout()); out_max->set_layout(x.layout());
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册