Skip to content

【Infer Symbolic Shape No.122】【BUAA】Add batch_fc - #67348

Merged
luotao1 merged 3 commits into
PaddlePaddle:developfrom
Whsjrczr:api4
Aug 16, 2024
Merged

luotao1 merged 3 commits into
PaddlePaddle:developfrom
Whsjrczr:api4

Conversation

@Whsjrczr

Copy link
Copy Markdown
Contributor

PR Category

CINN

PR Types

Others

Description

Add BatchFcOpInferSymbolicShape

@paddle-bot

paddle-bot Bot commented Aug 12, 2024

Copy link
Copy Markdown

你的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.

@paddle-bot paddle-bot Bot added the contributor External developers label Aug 12, 2024
@luotao1 luotao1 added the HappyOpenSource Pro 进阶版快乐开源活动,更具挑战性的任务 label Aug 13, 2024
Comment on lines +252 to +269
PADDLE_ENFORCE_EQ(
input_dims[0],
w_dims[0],
common::errors::InvalidArgument(
"Input.dim[0] and W.dim[0] of BatchFcOp should be same."));
PADDLE_ENFORCE_EQ(
input_dims[2],
w_dims[1],
common::errors::InvalidArgument(
"Input.dim[2] and W.dim[1] of BatchFcOp should be same."));
PADDLE_ENFORCE_EQ(bias_dims[0],
input_dims[0],
common::errors::InvalidArgument(
"Bias.dim[0] should be same as input.dim[0]."));
PADDLE_ENFORCE_EQ(bias_dims[1],
w_dims[2],
common::errors::InvalidArgument(
"Bias.dim[1] should be same as input.dim[2]."));

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

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

修改为添加约束

Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

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

收到

@gongshaotian gongshaotian left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

LGTM

@Whsjrczr

Copy link
Copy Markdown
Contributor Author
截屏2024-08-16 10 02 14 单测存在,pir测试未打开

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