Skip to content

[ASCEND][TLE]: add AddRmsNormBias triton kernel#641

Open
small-cat wants to merge 6 commits into
flagos-ai:triton_v3.2.xfrom
small-cat:wuzhenyu.dev
Open

[ASCEND][TLE]: add AddRmsNormBias triton kernel#641
small-cat wants to merge 6 commits into
flagos-ai:triton_v3.2.xfrom
small-cat:wuzhenyu.dev

Conversation

@small-cat
Copy link
Copy Markdown
Collaborator

add add_rmsnorm_bias triton kernels in python/tutorials/tle

@small-cat small-cat requested a review from zhzhcookie as a code owner June 2, 2026 06:24
@small-cat small-cat changed the title Wuzhenyu.dev [ASCEND][TLE]: add AddRmsNormBias triton kernel Jun 2, 2026
@zhzhcookie
Copy link
Copy Markdown
Collaborator

Please add this kernel to the ci workflow.

# flagtree tle tutorials
pushd python/tutorials/tle
pytest -s -k "accuracy" 07-add-rms-norm-bias.py
popd
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.

This branch has conflicts that must be resolved, please fix it.

@zhzhcookie
Copy link
Copy Markdown
Collaborator

Please run pre-commit run --all-files to pass the Code-Format-Check CI.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants