Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
PR types
Others
PR changes
Others
Description
Pcard-71699
框架支持stride后。发现split性能下降。
在split后调用matmul,假如split 3份,且axis不为0,此时前向需要调用3次contiguous,反向也调用3次contiguous。导致模型性能下降。
通过Python手动调用contiguous。可以将contiguous的次数降为3次。但3个contiguous kernel的性能不及1个split的性能。
在V100上,3个contiguous kernel比1个split慢7%。但在A100上慢87%。
#56866 尝试加速contiguous。但发生CUDA700,应该没有写对。即使写对了,新的写法速度不如原有写法。
暂时让split不走stride。