2026/6/20 11:01:04
网站建设
项目流程
做电子商务的网站,微信公众号服务号网站开发流程,中国上海门户网,订餐网站建设在开发gemm或attention之类的算子时#xff0c;通常面临着因规模过大造成的溢出问题#xff0c;体现为虽然单个数据的值很小#xff0c;但由于大规模的MMA操作#xff0c;导致结果变成inf。在进行测试时#xff0c;为覆盖不同的场景同时方便调试#xff0c;设置值为全1和…在开发gemm或attention之类的算子时通常面临着因规模过大造成的溢出问题体现为虽然单个数据的值很小但由于大规模的MMA操作导致结果变成inf。在进行测试时为覆盖不同的场景同时方便调试设置值为全1和idx是两种通常的情况但在大规模MMA中需要注意设置一定的缩放例如可以将全1变成0.001f。最好不要用idx或者改成某个周期内的idx因为idx会随着问题规模的增大而变大导致后面的结果虽然通过了缩放但仍然是很大的情况。血泪教训debug了很久发现是因为测试数据生成不合适造成的问题悲乎