[Auto Parallel] Add co_shard spmd_rule for bmm by ooooo-create · Pull Request #75555 · PaddlePaddle/Paddle
Conversation
PR Category
Auto Parallel
PR Types
New features
Description
Add co_shard spmd_rule for bmm，直接调用了 matmul 的接口，"bmk,bkn->bmn"
| @@ -0,0 +1,27 @@ | |||
| /* Copyright (c) 2024 PaddlePaddle Authors. All Rights Reserved. | |||
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
2024 -> 2025
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
好的，我下一个 pr 进行修改～
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters. Learn more about bidirectional Unicode characters