[mlir][vector] Fix extract op canonicalization for 0d vector
authorThomas Raoux <thomasraoux@google.com>
Tue, 17 Jan 2023 17:25:51 +0000 (17:25 +0000)
committerThomas Raoux <thomasraoux@google.com>
Tue, 17 Jan 2023 17:25:51 +0000 (17:25 +0000)
commitb015fccbe503fd7109405decb4f3eb6269e7706b
treee83be0be805bded9ada4342c3cf3a9d39c80431a
parent794979ad8cc975b4417dcf5fcce9c9e8e28e52b3
[mlir][vector] Fix extract op canonicalization for 0d vector

Fix ExtractOpFromBroadcast when the broadcast source is a 0d vector.

Differential Revision: https://reviews.llvm.org/D141735
mlir/lib/Dialect/Vector/IR/VectorOps.cpp
mlir/test/Dialect/Vector/canonicalize.mlir