Skip to content

Commit 1e89a76

Browse files
authored
[MLIR] Refactor to create vectorization convOp precondition check (#130181)
In corner situations, the vectorization pass may face to lower a conv2d op and assert in a completely irrelevant location in vectorizeConvolution() subroutine. ~~This PR rejects the conv2d op early and make the asserted routine to return failure as a defensive workaround.~~ In addressing this, the PR moved all condition check away from the `Conv1dGenerator` into the `convOpPreconditionCheck()` function. This makes the unsupported ops such as conv2d to be rejected early and leave a cleaner `Conv1dGenerator` constructor.
1 parent b79d53c commit 1e89a76

File tree

2 files changed

+219
-131
lines changed

2 files changed

+219
-131
lines changed

0 commit comments

Comments
 (0)