The GitHub Actions job "CI" on tvm.git/fix/relax-onnx-prelu-slope-broadcast has failed. Run started by GitHub user siyiweigeHEW (triggered by siyiweigeHEW).
Head commit for run: 236ab026ed3f6922a34fa1dcb57d25bfdbd7a9c4 / FFChopon <[email protected]> [Relax][Frontend][ONNX] Support broadcastable multi-axis PRelu slopes PRelu._impl_v1 only accepted a slope with a single non-broadcast axis. A valid ONNX PRelu whose slope is broadcastable to x across multiple axes (e.g. a slope shaped like x, or with several non-broadcast dims) was rejected with ValueError 'Invalid PRelu slope shape (multiple non-broadcast dims)'. onnxruntime and onnx.reference both accept and run such models correctly. nn.prelu can only express a single per-axis slope, so lower the multi-axis case elementwise as PRelu(x, s) = where(x < 0, s * x, x) via relax.op.where/less/multiply. The single non-broadcast axis and all-ones/slope-of-rank-1 cases keep using nn.prelu unchanged. Together with #20115 (lower-rank and rank-0 slopes), all unidirectionally broadcastable slope shapes accepted by onnxruntime now import. Co-Authored-By: Claude <[email protected]> Report URL: https://github.com/apache/tvm/actions/runs/32067602550 With regards, GitHub Actions via GitBox --------------------------------------------------------------------- To unsubscribe, e-mail: [email protected] For additional commands, e-mail: [email protected]
