-
Notifications
You must be signed in to change notification settings - Fork 5.8k
[DCU] fix fused_bias_act op #65399
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
[DCU] fix fused_bias_act op #65399
Conversation
你的PR提交成功,感谢你对开源项目的贡献! |
PR-CI-Hygon-DCU-Test失败单测中不包含 |
@qili93 已按照review修改,PR-CI-Hygon-DCU-Test失败原因非本算子单测 |
@@ -94,7 +94,8 @@ def fused_act_bias_wrapper( | |||
|
|||
|
|||
@unittest.skipIf( | |||
not core.is_compiled_with_cuda(), "core is not compiled with CUDA" | |||
not core.is_compiled_with_cuda() and not core.is_compiled_with_rocm(), | |||
"core is not compiled with CUDA or ROCm", |
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.
这里可以不用修改,你可以看下 is_compiled_with_cuda 的实现,当初了为支持HIP和CUDA的兼容,当编译的时候是HIP的时候,也会返回True,可以参考
Paddle/paddle/fluid/pybind/pybind.cc
Line 265 in 9eb5b03
#if !defined(PADDLE_WITH_CUDA) && !defined(PADDLE_WITH_HIP) |
下面几行也是一样。
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
PR Category
Custom Device
PR Types
Bug fixes
Description
card-84892
fix fused_bias_act_utils on DCU
fix load_store_util on DCU