在提交新问题之前,请确保您已经在社区中搜索过相关问题,并使用了社区中提供的资源/工具后,仍未找到满意的解决方式。
⚠️ 安全信息提醒:请仔细检查提供的文本内容,确保其不包含敏感数据信息,包括但不限于:
在分享配置信息或代码示例时,请将敏感信息脱敏处理,或使用 <TOKEN> 等占位符替代原有内容。
<TOKEN>
例如: - 操作系统 - 昇腾硬件信息 - CANN软件版本 - 安装的对应软件版本
修复rms_norm_quant_v2接口rstd的shape推导。当前rstd的正常推导只支持x.shape=[N,D],gamma.shape = [D]。修改后还支持x.shape=[N,D],gamma.shape = [1,D]
欢迎加入社区,感谢您对社区的贡献 🎉!
在提交新问题之前,请确保您已经在社区中搜索过相关问题,并使用了社区中提供的资源/工具后,仍未找到满意的解决方式。
⚠️ 安全信息提醒:请仔细检查提供的文本内容,确保其不包含敏感数据信息,包括但不限于:
在分享配置信息或代码示例时,请将敏感信息脱敏处理,或使用
<TOKEN>等占位符替代原有内容。环境信息
🐛 问题描述
修复rms_norm_quant_v2接口rstd的shape推导。当前rstd的正常推导只支持x.shape=[N,D],gamma.shape = [D]。修改后还支持x.shape=[N,D],gamma.shape = [1,D]
欢迎加入社区,感谢您对社区的贡献 🎉!