[mlir][vector] Fix contraction op lowering with mixed types
authorThomas Raoux <thomasraoux@google.com>
Thu, 16 Jun 2022 00:40:58 +0000 (00:40 +0000)
committerThomas Raoux <thomasraoux@google.com>
Thu, 16 Jun 2022 16:40:56 +0000 (16:40 +0000)
commitf011d32c3a625eb86d1e33a70100b0a031f5fcd4
treeb4cbb619437397d6c8d77e9d6c65439abec8a005
parenta70b39abffb42eb85fd2dd7960ec9dc7ae38510c
[mlir][vector] Fix contraction op lowering with mixed types

contraction op can have mixed type, add support for this case to the pattern
lowering contraction op to outerproduct.

Differential Revision: https://reviews.llvm.org/D127926
mlir/lib/Dialect/Vector/Transforms/VectorTransforms.cpp
mlir/test/Dialect/Vector/vector-contract-transforms.mlir