【AscendC】ReduceSum中指定workLocal大小时如何计算

参考官方文档,但其中对于workLocal大小的计算只是以代码注释的方式给出,不具有直观性,现在结合两个官方例子进行推导。

首先注意到api的调用,分为高维切分模式和指定数量模式,两种模式的计算是不同的,主要在于repeattimes这个变量的大小。

高维切分的例子见上图,可以看到结果是80.

结合图1进行分析,首先数据类型为half类型,所以elementsPerBlock为16,而repeatTime的计算是通过srcDataSize / mask, 也即8320 / 128 = 65。

进行上取整得到的结果是65 + 15 / 16 = 5, 5 * 16 = 80。所以最终结果为80。

对于指定数量n的模式,见下图:

此处指定的srcDataSize是288,由图1的注释,

// 此处需要注意:对于tensor高维切分计算接口,firstMaxRepeat就是repeatTimes;对于tensor前n个数据计算接口,firstMaxRepeat为count/elementsPerRepeat,比如在half类型下firstMaxRepeat就是count/128,在float类型下为count/64,按需填入,对于count<elementsPerRepeat的场景,firstMaxRepeat就是1

可以看到 firstMaxRepeated = 288 / 128 = 1。

因此得到的结果就是(1+ 15) / 16 * 16 = 16.

相关推荐
147API7 分钟前
评测分数不理想,什么时候值得补蒸馏数据
人工智能·深度学习·机器学习
不会写代码的女程序猿9 分钟前
养生馆采购 AI 四诊仪,5000 元预算怎么选?
大数据·人工智能·科技·ai·健康医疗
天天代码码天天21 分钟前
一个 HTML 就能跑完整 OCR:lw.PPOCR.C 发布 v0.1.0-preview.4,新增浏览器 JavaScript SDK
人工智能
AI情绪识别开源21 分钟前
检信 AI 智能推广平台(代号:JX-Promote)
人工智能·算法·erlang
俊哥V29 分钟前
每日 AI 研究简报 · 2026-08-30
人工智能·ai
Java后端的Ai之路29 分钟前
14、Python - 责任链模式
服务器·开发语言·人工智能·python·责任链模式
番茄不是西红柿kk32 分钟前
GLM-5.3-Flash 20分钟复刻《我的世界》实录
人工智能·ai·aigc·agent·我的世界
霸道流氓气质36 分钟前
Tabbit AI 原生浏览器 — 完全技术参考手册
人工智能
Python大数据分析@36 分钟前
推荐一个生成“结构化”html网页的SKILL
人工智能
阿里云大数据AI技术37 分钟前
DataWorks Data Agent 实战课堂(六):数据集成定时任务巡检
人工智能·agent