未验证 提交 748d2ae0 编写于 作者: S sneaxiy 提交者: GitHub

inplace addto (#42313)

上级 37e2f027
...@@ -178,9 +178,11 @@ void InplaceAddToOpPass::Run(Graph *graph) const { ...@@ -178,9 +178,11 @@ void InplaceAddToOpPass::Run(Graph *graph) const {
auto *out_generated_op = dynamic_cast<details::ComputationOpHandle *>( auto *out_generated_op = dynamic_cast<details::ComputationOpHandle *>(
out_var_ptr->GeneratedOp()); out_var_ptr->GeneratedOp());
// NOTE(zhiqiu): currently, only conv2d_grad supports addto strategy // FIXME(zengjinle): the "custom_fused_dense_grad" is only used for
// MLPerf temporarily. Replace it with the formal op type in the future.
if (right_generated_op->Name() != "conv2d_grad" && if (right_generated_op->Name() != "conv2d_grad" &&
right_generated_op->Name() != "resnet_unit_grad") { right_generated_op->Name() != "resnet_unit_grad" &&
right_generated_op->Name() != "custom_fused_dense_grad") {
continue; continue;
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册