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

【BUAA】【Infer Symbolic Shape】add shardindex #67407

Merged
merged 3 commits into from
Aug 16, 2024

Conversation

uanu2002
Copy link
Contributor

PR Category

CINN

PR Types

Others

Description

添加 ShardIndex 算子符号推导接口实现。

Copy link

paddle-bot bot commented Aug 14, 2024

你的PR提交成功,感谢你对开源项目的贡献!
请关注后续CI自动化测试结果,详情请参考Paddle-CI手册
Your PR has been submitted. Thanks for your contribution!
Please wait for the result of CI firstly. See Paddle CI Manual for details.

@luotao1 luotao1 added contributor External developers HappyOpenSource Pro 进阶版快乐开源活动,更具挑战性的任务 labels Aug 14, 2024
@uanu2002
Copy link
Contributor Author

shardindex存在对应单测,check_pir默认关闭,请求豁免coverage

Comment on lines 1770 to 1779
int64_t last_dim =
static_cast<int>(in_shape[in_shape.size() - 1].Get<std::int64_t>());
if (last_dim > 0) {
PADDLE_ENFORCE_EQ(last_dim,
1,
common::errors::InvalidArgument(
"The last dimension of Input(X) should be 1, "
"but the value given is %d.",
last_dim));
}
Copy link
Contributor

Choose a reason for hiding this comment

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

last_dim不一定就是int64_t,这里直接加约束就好

Hongqing-work
Hongqing-work previously approved these changes Aug 15, 2024
@luotao1
Copy link
Contributor

luotao1 commented Aug 16, 2024

冲突了

Copy link
Contributor

@gongshaotian gongshaotian left a comment

Choose a reason for hiding this comment

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

LGTM

@gongshaotian
Copy link
Contributor

麻烦补充描述下单测情况,并在表格中勾选

@uanu2002
Copy link
Contributor Author

麻烦补充描述下单测情况,并在表格中勾选

shardindex存在对应单测,check_pir默认关闭,请求豁免coverage

@luotao1 luotao1 merged commit 3646aef into PaddlePaddle:develop Aug 16, 2024
27 of 28 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
contributor External developers HappyOpenSource Pro 进阶版快乐开源活动,更具挑战性的任务
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants