漏洞概述 在 项目中,存在一个关于比较操作(compare ops)的转换问题。具体来说,将 ONNX 的比较操作转换为 TOSA(Tensor Operations Standard Architecture)时,可能会出现问题。 影响范围 系统:s390x-Linux, amd64-Linux, amd64-Windows, amd64-macOS 构建状态:所有系统的构建状态均为“pending”或“Build Status”,表明这些系统可能受到影响。 模型动物园状态:所有模型的构建状态均为“Failed”,表明这些模型可能无法正常工作。 修复方案 1. 更新代码: - 提交 添加了将 ONNX 比较操作转换为 TOSA 的代码(#3489)。 - 提交 再次添加了将 ONNX 比较操作转换为 TOSA 的代码(#3489)。 2. 检查依赖: - 确保使用正确的 LLVM 和 MLIR 版本。 - 检查 中的依赖项版本。 3. 测试和验证: - 在更新代码后,进行全面的测试和验证,确保转换功能正常工作。 - 使用提供的示例代码进行测试,确保模型能够正确加载、计算和输出结果。 POC 代码 以下是用于测试 ONNX-MLIR 的示例代码: 输出示例: 总结 该漏洞涉及 ONNX 比较操作到 TOSA 的转换问题,影响多个系统和模型。修复方案包括更新代码、检查依赖和进行全面的测试验证。提供的 POC 代码可用于测试和验证修复效果。