llvm-project
56e0b6af - [mlir][affine] Fix crash in vectorizeAffineLoopNest test utility for reduction loops (#184617)

Commit
12 days ago
[mlir][affine] Fix crash in vectorizeAffineLoopNest test utility for reduction loops (#184617) The test utility function `testVecAffineLoopNest` called `isLoopParallel` with a `reductions` output parameter, which populates reduction descriptors when the loop performs a reduction. However, these descriptors were never added to `strategy.reductionLoops` before calling `vectorizeAffineLoopNest`. When the vectorizer then processed a loop with `iter_args`, it found no reduction descriptors in the strategy and hit an assertion failure. Fix by registering the reduction loop descriptors in the strategy before vectorization, matching what the production vectorizer code already does correctly. Fixes #128334
Author
Parents
Loading