-
Notifications
You must be signed in to change notification settings - Fork 3k
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
support qwen dpo fused kernel #9686
Conversation
Thanks for your contribution! |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
use_fused_rms_norm=training_args.use_fused_rms_norm, | ||
use_fused_rope=training_args.use_fused_rope, | ||
use_fused_linear=training_args.use_fused_linear, | ||
use_fused_dropout_add=training_args.use_fused_dropout_add, |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
可以用 llm meta config, set_config 避免复杂的赋值操作。也可以不用
Codecov ReportAll modified and coverable lines are covered by tests ✅
Additional details and impacted files@@ Coverage Diff @@
## develop #9686 +/- ##
===========================================
+ Coverage 52.34% 53.01% +0.66%
===========================================
Files 726 718 -8
Lines 117265 112297 -4968
===========================================
- Hits 61377 59529 -1848
+ Misses 55888 52768 -3120 ☔ View full report in Codecov by Sentry. |
PR types
PR changes
Description
dpo支持 fused kernel 配置