未验证 提交 d7cfee9b 编写于 作者: W wangchaochaohu 提交者: GitHub

Checkout point add (#28488)

* upgrade pass capability
上级 98dc11bb
...@@ -394,5 +394,5 @@ REGISTER_PASS_CAPABILITY(squared_mat_sub_fuse_pass) ...@@ -394,5 +394,5 @@ REGISTER_PASS_CAPABILITY(squared_mat_sub_fuse_pass)
.EQ("square", 0) .EQ("square", 0)
.EQ("elementwise_mul", 0) .EQ("elementwise_mul", 0)
.EQ("elementwise_sub", 0) .EQ("elementwise_sub", 0)
.EQ("fill_constant", 0) .EQ("fill_constant", 1)
.EQ("fusion_squared_mat_sub", 0)); .EQ("fusion_squared_mat_sub", 0));
...@@ -14,6 +14,7 @@ limitations under the License. */ ...@@ -14,6 +14,7 @@ limitations under the License. */
#include "paddle/fluid/operators/fill_constant_op.h" #include "paddle/fluid/operators/fill_constant_op.h"
#include <string> #include <string>
#include "paddle/fluid/framework/op_version_registry.h"
namespace paddle { namespace paddle {
namespace operators { namespace operators {
...@@ -143,3 +144,12 @@ REGISTER_OP_CPU_KERNEL(fill_constant, ops::FillConstantKernel<float>, ...@@ -143,3 +144,12 @@ REGISTER_OP_CPU_KERNEL(fill_constant, ops::FillConstantKernel<float>,
ops::FillConstantKernel<int>, ops::FillConstantKernel<int>,
ops::FillConstantKernel<bool>, ops::FillConstantKernel<bool>,
ops::FillConstantKernel<paddle::platform::float16>); ops::FillConstantKernel<paddle::platform::float16>);
REGISTER_OP_VERSION(fill_constant)
.AddCheckpoint(
R"ROC(
Upgrade fill_constant, add a new input [ValueTensor].
)ROC",
paddle::framework::compatible::OpVersionDesc().NewInput(
"ValueTensor",
"In order to support new feature tensor support of Value"));
...@@ -171,6 +171,6 @@ REGISTER_OP_CPU_KERNEL(gather_grad, ops::GatherGradientOpKernel<float>, ...@@ -171,6 +171,6 @@ REGISTER_OP_CPU_KERNEL(gather_grad, ops::GatherGradientOpKernel<float>,
ops::GatherGradientOpKernel<uint8_t>, ops::GatherGradientOpKernel<uint8_t>,
ops::GatherGradientOpKernel<int64_t>); ops::GatherGradientOpKernel<int64_t>);
REGISTER_OP_VERSION(gather) REGISTER_OP_VERSION(gather)
.AddCheckpoint(R"ROC(upgrad gather, add attribut [axis])ROC", .AddCheckpoint(R"ROC(upgrad gather, add a new input [Axis])ROC",
paddle::framework::compatible::OpVersionDesc().NewAttr( paddle::framework::compatible::OpVersionDesc().NewInput(
"axis", "Specify the axis of gather operation.", {})); "Axis", "Specify the axis of gather operation."));
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册