Google Git
Sign in
llvm / llvm-project / mlir / 0a2fbd8e6a7be3794657b9447e82fcd579c43abb
commit0a2fbd8e6a7be3794657b9447e82fcd579c43abb[log] [tgz]
authorHsiangkai Wang <hsiangkai.wang@arm.com>Wed Sep 17 09:10:52 2025 +0100
committerCopybara-Service <copybara-worker@google.com>Wed Sep 17 01:15:48 2025 -0700
tree78017b319b789731e1c4359dc1999b7497ed690c
parent74ece8ec21ca8ce6c3f608db50b72d967124b98c [diff]
[mlir][gpu][vector] Lower Vector dialect to GPU for element-wise ops only (#159091)

Current convertVectorToMMAOps starts from vector.contract and finds its
dependencies as the targets to convert. In GPU dialect, we have
gpu.subgroup_mma_elementwise operation. We should be able to lower
element-wise operations to GPU MMA operations without vector.contract.
This patch adds this case to the pattern.

GitOrigin-RevId: 46ad5408fc8e31e97f5b1191320ca449c00f4191
2 files changed
tree: 78017b319b789731e1c4359dc1999b7497ed690c
  1. benchmark/
  2. cmake/
  3. docs/
  4. examples/
  5. include/
  6. lib/
  7. python/
  8. test/
  9. tools/
  10. unittests/
  11. utils/
  12. .clang-format
  13. .clang-tidy
  14. CMakeLists.txt
  15. LICENSE.TXT
  16. Maintainers.md
  17. README.md
README.md

Multi-Level Intermediate Representation

See https://mlir.llvm.org/ for more information.