Pull Request

变更摘要

本 PR 旨在让 ATC 的 ONNX parser 层支持 INT4 类型，从而能够解析含 INT4 的 ONNX 模型并完成图转 OM。主要改动包括在 ge_onnx.proto 与 onnx_util.h 中新增 INT4 枚举值、在 onnx_util.cc 中建立 OnnxDataType::INT4 到 ge::DataType::DT_INT4 的类型映射，并在 OnnxConstantParser 中实现 INT4 常量数据的打包字节拷贝逻辑（兼容 raw_data 与 int32_data 两种存储方式），同时修复了 OnnxFileConstantParser 中对未定义数据类型的判断条件。此外补充了 UT 与 ST 测试用例，覆盖类型映射、数据对齐、padding 截断及越界保护等场景。

主要改动

• 新增 INT4 类型定义: 在 graph_metadef/proto/onnx/ge_onnx.proto 的 TensorProto.DataType 枚举中新增 INT4 = 22，并注明 4-bit 有符号整数、每字节打包 2 个值、低位在前；同时在 parser/parser/onnx/onnx_util.h 的 OnnxDataType 枚举中新增 INT4 = 22。

• 新增 INT4 类型映射: 在 onnx_util.cc 的 onnx_data_type_map 中增加 {OnnxDataType::INT4, ge::DataType::DT_INT4}，使 ONNX 的 INT4 能映射到 GE 的 DT_INT4。

• OnnxConstantParser 支持 INT4 常量数据: 在 onnx_constant_parser.cc 中为 INT4 增加基于 int32_data_size() 的数据量统计，并在 ParseConvertDataElements 中新增 INT4 分支，通过 ge::GetSizeInBytes 计算精确的打包字节数，按 min(byte_size, available) 从 int32_data 直接拷贝，避免 padding 槽位带来的多余数据及越界读取。

• 修正未定义数据类型的判断逻辑: 在 onnx_file_constant_parser.cc 的 ParseDataType 中，将 type >= ge::DataType::DT_UNDEFINED 改为 type == ge::DataType::DT_UNDEFINED，避免将合法的 DT_INT4 等类型误判为未定义。

• 补充测试用例: 新增 UT 与 ST 用例及 onnx_int4_gen.py 生成脚本，覆盖 ParseConvertDataType_Int4、raw_data 路径、int32_data 对齐、奇数个数截断、数据不足防越界，以及端到端解析含 INT4 Constant 的 ONNX 模型并校验 DT_INT4 类型与打包字节内容。

代码审查

✅ 未发现问题

/lgtm

/approve