缺陷信息

缺陷描述

在TilingDiagFlat函数中，第108-109行使用sizeof(dataType)计算auxMatrix和outputDataSize，其中dataType是ge::DataType枚举类型，sizeof(dataType)返回枚举类型自身大小（通常4字节），而非该枚举值所代表的数据类型大小（如float16为2字节、float64为8字节）。正确方式应使用ge::GetSizeByDataType(dataType)，同文件第43行和第64行即采用了该正确方式。对于8字节数据类型，auxMatrix和outputDataSize被低估为实际值的1/2，导致inputDataSize=ubSize-auxMatrix-outputDataSize被高估，内核可能认为UB可容纳更多输入数据而导致UB缓冲区溢出。

事实核查

经复核源代码确认，TilingDiagFlat函数（第87-138行）第108-109行使用sizeof(dataType)计算auxMatrix和outputDataSize，其中dataType为ge::DataType枚举类型，sizeof(dataType)返回枚举类型大小（4字节），而非该枚举值代表的数据类型实际大小。

同文件CalcScalarTiling（第43行）和CalcAuxMatrixTiling（第64行）均正确使用ge::GetSizeByDataType(dataType)，形成鲜明对比。

对8字节数据类型（如DT_FLOAT64、DT_INT64），sizeof版本将auxMatrix/outputDataSize低估为实际值的1/2，导致inputDataSize=ubSize-auxMatrix-outputDataSize被高估，下游内核可能向UB写入超出实际容量的数据导致溢出。

调用链已验证：Tiling4DiagV2(diag_v2_tiling.cpp:86)->processDiagFlat(diag_v2_tiling.cpp:71->74)->TilingDiagFlat(diag_flat_tiling.cpp:87)，通过IMPL_OP_OPTILING注册为tiling入口，可达。

数据流证据

Source（问题源头）

conversion/diag_v2/op_host/arch32/diag_v2_tiling.cpp:86 行 Tiling4DiagV2 作为注册的tiling入口函数接收 gert::TilingContext* context

Sink（问题爆发点）

conversion/diag_flat/op_host/arch32/diag_flat_tiling.cpp:108 行 auto auxMatrix = SCALAR_THRESHOLD_NUM * SCALAR_THRESHOLD_NUM * sizeof(dataType) 使用sizeof(枚举)而非数据类型大小，导致auxMatrix/outputDataSize计算错误

传播路径：

调用链

可达调用链1　起点：(entry functions) → 终点：optiling::TilingDiagFlat　深度：2

修复建议

auto auxMatrix = SCALAR_THRESHOLD_NUM * SCALAR_THRESHOLD_NUM * ge::GetSizeByDataType(dataType);
auto outputDataSize = SCALAR_THRESHOLD_NUM * SCALAR_THRESHOLD_NUM * ge::GetSizeByDataType(dataType);