-
Notifications
You must be signed in to change notification settings - Fork 5.6k
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
[2/4] CUDNNv8 ResNet Fusion: Add fused_scale_bias_add_relu OP #58504
[2/4] CUDNNv8 ResNet Fusion: Add fused_scale_bias_add_relu OP #58504
Conversation
|
Sorry to inform you that 09f5aff's CIs have passed for more than 7 days. To prevent PR conflicts, you need to re-run all CIs manually. |
8af360a
to
99c5d30
Compare
- op : fused_scale_bias_add_relu | ||
args : (Tensor x1, Tensor scale1, Tensor bias1, Tensor x2, Tensor scale2, Tensor bias2, bool fuse_dual, bool exhaustive_search) | ||
optional : scale2, bias2 | ||
output : Tensor(y) |
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.
一般来说,单输出都取名为out
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.
会在下个PR修改.
fuse_dual)); | ||
} | ||
// set output dims | ||
y->set_dims(x1.dims()); |
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.
严格来说,这里也需要设置dtype和layout
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.
会在下个PR修改.
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 check_dygaph =False
@zyfncg Hi, would you please merge? |
…Paddle#58504) * Rename op * Add fused_scale_bias_add_relu
…Paddle#58504) * Rename op * Add fused_scale_bias_add_relu
PR types
New features
PR changes
OPs
Description
This PR adds fused_scale_bias_add_relu op, which is part of the CUDNNv8 ResNet fusion integration.