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

【prim】update Prim optest about customvjp test, add FLAGS_comp_skip_default_ops #71269

Closed

Conversation

xiaoguoguo626807
Copy link
Contributor

@xiaoguoguo626807 xiaoguoguo626807 commented Feb 25, 2025

PR Category

Execute Infrastructure

PR Types

Improvements

Description

pcard-67164

增加前向拆解默认跳过的list 包含 embedding, dropout 算子,原因是编译器不好处理embedding 拆解得到的gather 等算子, dropout 拆解的uniform和原始dropout的随机数生成无法对齐。

添加跳过前向拆解默认算子的flag, FLAGS_comp_skip_default_ops=True, 默认为true 表示默认不拆解上述算子,如需取消可以设置为false 与反向拆解FLAGS_prim_vjp_skip_default_ops 对应

修改prim_op_test 反向测试的逻辑,如果是custom_vjp 则反向需要跳过相应的前向算子拆解。 之前测试的是前向拆解结果自动微分得到的结果,修改为使用反向拆解的结果。

Copy link

paddle-bot bot commented Feb 25, 2025

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

@CLAassistant
Copy link

CLA assistant check
Thank you for your submission! We really appreciate it. Like many open source projects, we ask that you all sign our Contributor License Agreement before we can accept your contribution.
1 out of 2 committers have signed the CLA.

✅ xiaoguoguo626807
❌ wangruting


wangruting seems not to be a GitHub user. You need a GitHub account to be able to sign the CLA. If you have already a GitHub account, please add the email address used for this commit to your account.
You have signed the CLA already but the status is still pending? Let us recheck it.

@phlrain phlrain self-requested a review February 28, 2025 04:35
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.

4 participants