未验证 提交 c037453d 编写于 作者: C Chitsing KUI 提交者: GitHub

fix mea get pad no default return bug (#54644)

上级 182e0904
......@@ -91,6 +91,8 @@ phi::DenseTensor get_pad_lse(const phi::GPUContext& dev_ctx,
ViewSliceHelper<T><<<grid, block, 0, dev_ctx.stream()>>>(
in_data, stride, in_dim[2], out_second_dim);
return *lse;
} else {
return *lse;
}
}
} // namespace funcs
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册