Skip to content
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

[OpenCL][ut_bugs]fix_opencl_split_bugs test=develop #8322

Merged
merged 3 commits into from
Jan 24, 2022

Conversation

zhenlin-work
Copy link
Collaborator

@zhenlin-work zhenlin-work commented Jan 21, 2022

当前修复原有kernel的diff并且扩展支持输入1 2 3 4维,对应axis=0 1 2 3,唯一的限制是输出tensor个数仍为2,是否需要支持多个tensor的split输出需要后续讨论必要性。

@paddle-bot-old
Copy link

Thanks for your contribution!

Copy link
Collaborator

@zhaoyang-star zhaoyang-star left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

加上一个判断吧,输出tensor个数不为2时,LOG(FATAL)报错退出。

Copy link
Collaborator

@zhaoyang-star zhaoyang-star left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@zhenlin-work zhenlin-work merged commit 297b309 into PaddlePaddle:develop Jan 24, 2022
WeiLi233 pushed a commit to WeiLi233/Paddle-Lite that referenced this pull request Mar 29, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants