The GitHub Actions job "Lint" on tvm.git/fix/onnx-hardmax-symbolic-axis has failed. Run started by GitHub user LiRunGuo (triggered by LiRunGuo).
Head commit for run: e0c85927570bff4a36dc33fb9c19a1a6c8b2721e / RunguoLi <[email protected]> [Fix][Relax][ONNX] Support Hardmax over a symbolic axis extent Hardmax is lowered to argmax followed by one_hot, but one_hot takes its depth as a static integer attribute. When the reduced axis has a symbolic extent the importer passed a tir.Var (or, for opset <= 12, the symbolic product of the flattened dims) and failed with a TypeError. Keep the one_hot lowering for static extents, and for symbolic extents compare the keepdims argmax against an arange over the axis broadcast along it. argmax returns the first maximum, so ties still resolve to the first index as ONNX requires. Report URL: https://github.com/apache/tvm/actions/runs/35747596957 With regards, GitHub Actions via GitBox --------------------------------------------------------------------- To unsubscribe, e-mail: [email protected] For additional commands, e-mail: [email protected]
