Skip to content

【Fix PIR Unittest No.76,78,79,80】Fix some test cast in PIR - #64454

Merged
wanghuancoder merged 1 commit into
PaddlePaddle:developfrom
wanghuancoder:fix_pir_api_6
May 21, 2024
Merged

wanghuancoder merged 1 commit into
PaddlePaddle:developfrom
wanghuancoder:fix_pir_api_6

Conversation

@wanghuancoder

Copy link
Copy Markdown
Contributor

PR Category

Execute Infrastructure

PR Types

Bug fixes

Description

Pcard-67164
默认开启PIR模式后,修复5个单测:
test_conv2d_api.py:使用了paddle.static.nn.conv2d
test_dist_fleet_a_sync_optimizer_geo.py:使用了paddle.static.nn.fc
test_eager_run_program.py:强依赖老静态图特性
test_generator_dataloader.py:DataLoader.from_generator是废弃API。但本单测触发了softmax的一处调用错误。已修复。

关联Issue:#63740

@paddle-bot

paddle-bot Bot commented May 20, 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.


act_op = getattr(_C_ops, act)
if act == 'softmax':
return act_op(input, -1)

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.

-1 表示axix? 若是,这里为什么固定为-1?

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.

这个对齐原有就静态图,旧静态图没有传这个值,就会用默认值。

@wanghuancoder
wanghuancoder merged commit e2c9020 into PaddlePaddle:develop May 21, 2024
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.

5 participants