-
Notifications
You must be signed in to change notification settings - Fork 5.7k
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
[OneDNN][PIR] Add depthwise_conv_onednn_pass #63051
[OneDNN][PIR] Add depthwise_conv_onednn_pass #63051
Conversation
你的PR提交成功,感谢你对开源项目的贡献! |
787c75b
to
0e1c434
Compare
512a708
to
a559af7
Compare
这个op不支持python测的test case,只能用c++实现。但win32下test case编译会不过,且py3的CI中似乎并没有默认开启MKLDNN,导致链接会有问题,因此暂时在Make file中过滤了win32 |
a559af7
to
92813b1
Compare
|
||
} // namespace pir | ||
|
||
REGISTER_IR_PASS(depthwise_conv_mkldnn_pass, DepthwiseConvMKLDNNPass); |
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.
pass命名中的mkldnn和onednn是否可以统一成onednn?
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.
好的~
@@ -0,0 +1,69 @@ | |||
// Copyright (c) 2023 PaddlePaddle Authors. All Rights Reserved. |
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.
使用这种方式写单测的原因是,组网时没法组出depthwise_conv算子吗?
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.
是的,python 端没有depthwise_conv的api支持,现有pir的test case架构应该暂时不支持这种,只能走c++侧
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.
是可以的,已改,多谢~
32906b2
to
f5ebb8f
Compare
test/ir/pir/fused_pass/onednn/test_depthwise_conv_onednn_pass.py
Outdated
Show resolved
Hide resolved
b122afb
to
2597663
Compare
* first commit of depthwise conv pass * style fix * add copy_onnx * add other create to test * check if onednn pass not register * add ifdef PADDLE_WITH_DNNL * add WITH_MKLDNN * fix style bug * add PADDLE_WITH_DNNL * add condition in onnx * SKIP WIN32 CI * name change mkl to onednn * change name * use python ut for depthwise conv * delete skipif * Rename test_depthwise_conv_onednn_pass.py to test_pir_depthwise_conv_onednn_pass.py
* first commit of depthwise conv pass * style fix * add copy_onnx * add other create to test * check if onednn pass not register * add ifdef PADDLE_WITH_DNNL * add WITH_MKLDNN * fix style bug * add PADDLE_WITH_DNNL * add condition in onnx * SKIP WIN32 CI * name change mkl to onednn * change name * use python ut for depthwise conv * delete skipif * Rename test_depthwise_conv_onednn_pass.py to test_pir_depthwise_conv_onednn_pass.py
PR Category
Others
PR Types
New features
Description
We rewrite depthwise_conv_onednn_pass for PIR.