Skip to content

[WIP] optimize group_normfor ASCEND_NPU#1154

Draft
sunyi0505 wants to merge 2 commits intolinkedin:mainfrom
sunyi0505:main
Draft

[WIP] optimize group_normfor ASCEND_NPU#1154
sunyi0505 wants to merge 2 commits intolinkedin:mainfrom
sunyi0505:main

Conversation

@sunyi0505
Copy link
Copy Markdown

Summary

Testing Done

  • Hardware Type:
  • run make test to ensure correctness
  • run make checkstyle to ensure code style
  • run make test-convergence to ensure convergence

@sunyi0505 sunyi0505 changed the title [WIP] optimize group_norm for ASCEND_NPU [WIP] optimize group_norm_backward for ASCEND_NPU Mar 20, 2026
@sunyi0505 sunyi0505 marked this pull request as draft March 20, 2026 08:01
@sunyi0505
Copy link
Copy Markdown
Author

sunyi0505 commented Mar 20, 2026

forward:
image
backward:
image

@sunyi0505 sunyi0505 changed the title [WIP] optimize group_norm_backward for ASCEND_NPU [WIP] optimize group_normfor ASCEND_NPU Mar 24, 2026
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.

1 participant