已合并
FlatQuant修复大shape内存检测问题 #3989
陈海杰创建于 4月20日
FlatQuant修复大shape内存检测问题 #3989
已合并
陈海杰创建于 4月20日
陈海杰成员
4月20日

描述

FlatQuant修复大shape内存检测问题

关联的Issue

https://gitcode.com/cann/ops-nn/issues/2172

测试

用单算子OOM包验证1400条泛化用例,功能和精度全部OK

文档更新

不涉及

类型标签

  • Bug修复
  • 新特性
  • 性能优化
  • 文档更新
  • 其他,请描述:
likedislike
Pull Request已成功合入, 合并人@CANN-robot
(感谢 陈海杰 的贡献)
陈海杰成员
4月20日 创建了 pull request,commit 0da5abca
CANN-robot
CANN-robot成员
4月20日 评论:

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
quant Chen_HaoWen, liujie12345678 (2/2) Chen_HaoWen, liujie12345678 (2/1)

💡 Tip:

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

CLA Signature Pass

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

likedislike
CANN-robotCANN-robot成员
4月20日 添加了label:cann-cla/yes
CANN-robotCANN-robot成员
4月20日 将sxb154714,chenfeng61,crystalhu,yangyang016,fanqirui,zhou-qilong,zhajianqing123,chenqi317,liubo75,tangweiwei2,wangzitao_leo,liujie12345678,Chen_HaoWen,wangyongguang,zhang-wu设为评审人
CANN-robotCANN-robot成员
4月20日 将sxb154714,chenfeng61,crystalhu,zhou-qilong,zhajianqing123,chenqi317,liubo75,tangweiwei2,wangzitao_leo,liujie12345678,Chen_HaoWen,wangyongguang,zhang-wu设为审查人
陈海杰成员
4月20日 评论:

compile

likedislike
陈海杰成员
4月20日 关联了issue:[Documentation|文档反馈]: 【Log】算子报错打印ERROR日志模块为slog
陈海杰成员
4月20日 删除了关联的issue:[Documentation|文档反馈]: 【Log】算子报错打印ERROR日志模块为slog
陈海杰成员
4月20日 关联了issue:[Bug-Report|缺陷反馈]: OOM包执行FlatQuant用例报错
CANN-robot
CANN-robot成员
4月20日 评论:

问题/功能描述

本次PR旨在优化两个核心算子内核(flat_quant_cube.hflat_quant_cube_one.h)的计算逻辑。原实现中,tiling()函数在计算invalidK变量时使用了浮点数除法,这不仅可能带来不必要的性能开销,还存在潜在的精度损失风险,可能影响特定形状参数下尾块判断的准确性。

修改方案描述

修改方案将tiling()函数中计算invalidK的表达式从static_cast<int64_t>(static_cast<float>(shape.K * shape.M - shape.Mceil) / static_cast<float>(shape.M))简化为(shape.K * shape.M - shape.Mceil) / shape.M。核心优化是移除了冗余的浮点数转换,直接使用整数除法进行计算。此举简化了代码逻辑,提升了计算效率,并从根本上避免了因浮点数转换可能导致的精度问题。

likedislike
CANN-robotCANN-robot成员
4月20日 添加了label:ci-pipeline-running
CANN-robot
CANN-robot成员
4月20日 评论:

流水线任务触发成功
任务链接 [4f3d1eaa97c94576b1646afad0c0cde6][流水线指导]

任务名称状态日志下载链接
Compile_Ascend_X86_ubuntu24 ✅ SUCCESS >>>>>
Compile_Ascend_X86_mobile_station ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_ARM ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_single ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_experimental ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_X86_950 ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_ARM_950 ✅ SUCCESS >>>>> >>>>>
Pre_Compile ✅ SUCCESS
pre_comment ✅ SUCCESS >>>>>
Compile_Ascend_X86 ✅ SUCCESS >>>>> >>>>>
Compile_Ascend_X86_mobile_station_ubuntu24 ✅ SUCCESS >>>>>
Compile_Ascend_X86_950_ubuntu24 ✅ SUCCESS >>>>>
Compile_Ascend_ARM_ubuntu24 ✅ SUCCESS >>>>>
Compile_Ascend_single_ubuntu24 ✅ SUCCESS >>>>>
Compile_Ascend_experimental_ubuntu24 ✅ SUCCESS >>>>>
Compile_Ascend_ARM_950_ubuntu24 ✅ SUCCESS >>>>>
Compile_X86_monitor ✅ SUCCESS >>>>> >>>>>
Compile_ARM_monitor ✅ SUCCESS >>>>> >>>>>
UT_Test_ophost ✅ SUCCESS
UT_Test_opapi ✅ SUCCESS
UT_Test_kernel ✅ SUCCESS
UT_Test_opgraph ✅ SUCCESS
Smoke_A900_npupool ✅ SUCCESS >>>>>
API_Check ✅ SUCCESS >>>>>
ATK_Test_A2 ✅ SUCCESS

[2026-04-20 14:54:01]    CI执行结束

likedislike
CANN-robot
CANN-robot成员
4月20日 评论:

流水线任务触发成功
任务链接 [dcdfd04c221d442baab5136616ac092a][流水线指导]

任务名称状态日志下载链接
codecheck ✅ SUCCESS >>>>>
anti_virus ✅ SUCCESS >>>>>
Check_Pr ✅ SUCCESS >>>>>
StaticCheck_codespell_check ✅ SUCCESS
StaticCheck_link_validity_check ✅ SUCCESS
StaticCheck_resource_existence_check ✅ SUCCESS
StaticCheck_tag_closed_check ✅ SUCCESS
StaticCheck_markdownlint ✅ SUCCESS
codecheck_style ✅ SUCCESS

[2026-04-20 14:51:49]    CI执行结束

likedislike
CANN-robotCANN-robot成员
4月20日 添加了label:api-check-pass
CANN-robotCANN-robot成员
4月20日 删除了label:ci-pipeline-running
CANN-robotCANN-robot成员
4月20日 添加了label:ci-pipeline-passed
陈海杰成员
4月20日 修改了pull request 的描述
Chen_HaoWen成员
4月20日 评论:

/approve

likedislike
CANN-robotCANN-robot成员
4月20日 添加了label:approved
liujie12345678成员
4月20日 评论:

/lgtm
/approve

likedislike
CANN-robotCANN-robot成员
4月20日 添加了label:lgtm
CANN-robot
CANN-robot成员
4月20日 评论:

Review Guide

This pull-request passes review.
Committers who wrote a comment of /approve are: Chen_HaoWen, liujie12345678.
Reviewers who wrote a comment of /lgtm are: liujie12345678, Chen_HaoWen.

likedislike
CANN-robotCANN-robot成员
4月20日 合入了pull request