【AscendC】tiling方案设计不当引起的一个时隐时现的bug

在设计tiling方案时,通常会考虑到非对齐的场景,对输入数据进行补全操作从而使得非对齐场景也能正确的完成计算。但在某些算子的实现过程中,沿用上述操作却会造成数据的错误计算,且这种错误出现与否取决于随机生成的测试数据质量。本文笔者记录一个典型的错误,并分析产生该错误的原因。

对于exp操作来说,通过将其优化为max + exp(x - max)是一种常用的手段。但对于非对齐场景,上述优化之后的算子可能会出现问题。对于补齐的位置,通常是以补0作为典型的补齐手段,此时如果原数据的最大值为大于等于0的值,那么在计算max的过程不会产生影响,但是在计算x-max之后就会造成补0的位置的值变成了负数,从而经过指数操作之后会产生正值,如果其后有规约求和的操作就会造成求和的数目过大从而造成结果错误。

但是上述过程可能会出现无法检测的情况,也即补齐的数目非常少(例如只有1个位置需要补0),此时可能出现没有影响到最终结果的情况。

因此在tiling设计中尤其要注意非对齐场景,这些补齐的位置是否会对计算产生影响。

相关推荐
Είναι η κοπέλα10 小时前
CUDA 与 N 卡驱动安装
人工智能·开源
TechEdu20260610 小时前
[人工智能]Python10:NumPy.random.Algebra 随机代数实战指南
人工智能·numpy
阿阿阿安10 小时前
Agent 智能体开发(二)Agent 经典架构范式构建
人工智能·ai·agent
AI你一生一世10 小时前
Attention is all you have:当上下文成为唯一的护城河
人工智能·大语言模型·注意力机制·rag·长上下文·上下文窗口·推理成本
天一生水water10 小时前
时间序列非平衡样本分类研究综述:发展脉络、核心方法与前沿方向
人工智能·分类·数据挖掘
Omics Pro10 小时前
预测性虚拟细胞中显式机理算子
大数据·数据库·人工智能·python·算法·机器学习·自然语言处理
haliu10 小时前
【FHE 同态加密】我们如何实现同态加密推理(十六):未完成清单与已知边界(同态加密推理的精度、性能与安全)
人工智能·嵌入式·c·fhe·推理引擎·c11·边缘推理·同态加密推理
2601_9659690610 小时前
教师选择AI认证,应该重点看哪些考核内容?
人工智能·数据挖掘
墨心@10 小时前
【无标题】
人工智能·大语言模型·agent·智能体·harness
康实训10 小时前
2026营养实训室建设标准与实施方案
人工智能·实训室·实训室建设