CARVIEW |
Navigation Menu
-
Notifications
You must be signed in to change notification settings - Fork 5.8k
New ir support fuse bn add act #56247
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
New ir support fuse bn add act #56247
Conversation
… new_ir_support_fuse_bn_add_act
…e/Paddle into new_ir_support_fuse_bn_add_act
你的PR提交成功,感谢你对开源项目的贡献! |
❌ The PR is not created using PR's template. You can refer to this Demo. |
… support_new_ir_load_combine
… support_new_ir_load_combine
… support_new_ir_load_combine
… new_ir_support_fuse_bn_add_act
…e/Paddle into new_ir_support_fuse_bn_add_act
… new_ir_support_fuse_bn_add_act
… new_ir_support_fuse_bn_add_act
… new_ir_support_fuse_bn_add_act
… new_ir_support_fuse_bn_add_act
paddle/phi/api/yaml/op_compat.yaml
Outdated
saved_variance: SavedVariance | ||
reserve_space: ReserveSpace | ||
|
||
- op : fused_bn_add_activation |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
为什么这里要添加两次映射?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
done
@@ -670,6 +670,10 @@ std::shared_ptr<paddle::framework::OperatorBase> BuildOperatorBase( | |||
|
|||
std::string fluid_op_name = op_yaml_info.GetOriginOpName(); | |||
|
|||
if (fluid_op_name == "fused_bn_add_activation_") { |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
上面的GetOriginOpName获取到的名字就是 yaml 中配置的fused_bn_add_activation,不需要再加这个判断了
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
done
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM for YAML
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM for add op input
* support new ir load combine * update * polish code * remove print * update * update * update * polish code * fix bug * polish code * fix compile bug * fix bug * revert code * remove useless code * polish code
PR types
Function optimization
PR changes
Others
Description
新IR支持fluid算子, fuse bn add activation;
该pr修改了fused_bn_add_activation_op的输入,新增Mean和Variance的输入,这两个输入是和MeanOut和VarianceOut共享同一片数据区,所以即使不新增,运行也不会错误,但是会存在一个问题,从IR fuse的角度,fuse出的目标子图,比源子图少了两个输入,这个是不符合安全性要求的规范的
Others
Pcard-67164