Skip to content

[NPU] Add support for the grpo_loss#1146

Merged
Tcc0403 merged 6 commits intolinkedin:mainfrom
UserChen666:grpo_loss
Mar 18, 2026
Merged

[NPU] Add support for the grpo_loss#1146
Tcc0403 merged 6 commits intolinkedin:mainfrom
UserChen666:grpo_loss

Conversation

@UserChen666
Copy link
Copy Markdown
Contributor

@UserChen666 UserChen666 commented Mar 13, 2026

Summary

Testing Done

image
  • Hardware Type: Atlas 800I A2
  • run make test to ensure correctness
  • run make checkstyle to ensure code style
  • run make test-convergence to ensure convergence

@UserChen666 UserChen666 changed the title [NPU]: Added support for the grpo_loss [NPU] Add support for the grpo_loss Mar 13, 2026
@UserChen666 UserChen666 force-pushed the grpo_loss branch 2 times, most recently from 8741d26 to d01d8e1 Compare March 15, 2026 06:16
@UserChen666
Copy link
Copy Markdown
Contributor Author

@Tcc0403 please review

Copy link
Copy Markdown
Collaborator

@Tcc0403 Tcc0403 left a comment

Choose a reason for hiding this comment

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

lgtm, could you attach testing results in the comment?

@UserChen666
Copy link
Copy Markdown
Contributor Author

UserChen666 commented Mar 17, 2026

image test cases passed @Tcc0403

Tcc0403
Tcc0403 previously approved these changes Mar 17, 2026
@Tcc0403 Tcc0403 enabled auto-merge March 17, 2026 12:14
auto-merge was automatically disabled March 18, 2026 02:43

Head branch was pushed to by a user without write access

@Tcc0403
Copy link
Copy Markdown
Collaborator

Tcc0403 commented Mar 18, 2026

please resolve conflict

@UserChen666
Copy link
Copy Markdown
Contributor Author

please resolve conflict

all resolved

Copy link
Copy Markdown
Collaborator

@Tcc0403 Tcc0403 left a comment

Choose a reason for hiding this comment

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

fix formatting issue

fix imports in __init__.py
@UserChen666
Copy link
Copy Markdown
Contributor Author

fix formatting issue

done.

@UserChen666
Copy link
Copy Markdown
Contributor Author

Something wrong with my IDE, so I made some mistakes when I git

@Tcc0403
Copy link
Copy Markdown
Collaborator

Tcc0403 commented Mar 18, 2026

No worries! It happens to me sometimes. Kind reminder, you can install pre-commit hook via pre-commit/prek for formatting and linting before commits. See: https://github.com/linkedin/Liger-Kernel/blob/main/docs/contributing.md#setting-up-your-development-environment

we should probably update our PR template to remind pre-commit hooks setup.

@Tcc0403 Tcc0403 added this pull request to the merge queue Mar 18, 2026
Merged via the queue into linkedin:main with commit ab7fe2a Mar 18, 2026
5 of 7 checks passed
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.

2 participants