已合并
[feat] add mxfp4_to_bf16_dequantization kernel #3624
yanghongru创建于 7月6日
[feat] add mxfp4_to_bf16_dequantization kernel #3624
已合并
yanghongru创建于 7月6日
yanghongru成员
7月6日

What this PR does / why we need it?

添加 MXFP4 反量化 BF16 的 triton 算子实现,用于 W4A16 的低精度 RL 后训练。

Does this PR introduce any user-facing change?

N/A

How was this patch tested?

在 950DT 单卡使用大小为(1024,4096)的矩阵测试耗时 0.1011ms。

正确性测试结果:https://wiki.huawei.com/domains/150199/wiki/301399/WIKI2026071911918935

likedislike
Pull Request已成功合入, 合并人@ascend-robot
(感谢 yanghongru 的贡献)
Yyanghongru成员
7月6日 创建了 pull request,commit 3f96163b
ascend-robotascend-robot成员
7月6日 添加了label:ascend-cla/no
ascend-robot
ascend-robot成员
7月6日 评论:

CLA Signature Guide

@yhr6385 , thanks for your pull request.

The following commit(s) have not signed Contributor License Agreement (CLA).

Commit Reason
f1a4c2f4 add mxfp4_to_bf16_dequantization... it's an invalid email! please check it.

If you need to sign CLA, you can click here to sign the CLA.

If you need to check if the email is set up correctly, you can click here to do it in the FAQs.

After signing the CLA or updating the email, you must comment /check-cla to check the CLA status again.

likedislike
ascend-robot
ascend-robot成员
7月6日 评论:

Thanks for your pull-request.
The full list of commands accepted by me can be found at here
You can get sig-info at here


PR Approval Progress

Congratulations! All modules have met the lgtm and approve requirements.

Module Approval Details

module lgtm status approve status
repo-Ascend/MindSpeed 郭鹏, 王铭震 (2/2) 郭鹏 (1/1)

💡 Tip:

  • Committer can comment /approve or /lgtm
  • Commenting /approve implies both code review (lgtm) and intent to merge (approve)

CLA Signature Pass

yhr6385, thanks for your pull request. All authors of the commits have signed the CLA. 👍

likedislike
此处折叠了120条消息 查看更多
mingzhenwang成员
8月3日 评论:

/lgtm

likedislike
gp513成员
8月3日 评论:

/approve

likedislike
ascend-robotascend-robot成员
8月3日 添加了label:approvedlgtm
ascend-robotascend-robot成员
8月3日 合入了pull request
ascend-robot
ascend-robot成员
8月3日 评论:

Pull Request 已合并或已关闭。

If you want to solve this problem, you can click here to do it in the FAQs.

likedislike