Google Git
Sign in
llvm / llvm-project / mlir / 7edb68990f02c2f650bcce8d3aae50dbdc919375
commit7edb68990f02c2f650bcce8d3aae50dbdc919375[log] [tgz]
authorPetr Kurapov <petr.a.kurapov@intel.com>Wed Oct 02 17:23:55 2024 +0200
committerCopybara-Service <copybara-worker@google.com>Wed Oct 02 08:28:57 2024 -0700
treeb7e602d30fa8c72be9ae8d82859c7ed31ac64b41
parent727bb98631e5b6650457dc17bcb6625f8a987abc [diff]
[MLIR][XeGPU] Add sg_map attribute to support Work Item level semantics (#108864)

The PR adds an attribute (sg_map) describing the distribution of
computation among work items for xegpu operations to be used in lowering
passes. The map is attached to the tensor descriptor, so the constructor
and the type are updated. Tests check the custom parser & printer. The
attribute is optional now, so no other changes required.
The complete description of the attribute can be found
[here](https://github.com/intel/mlir-extensions/blob/main/docs/rfcs/XeGPU.md#xegpu-attributes-to-support-work-item-level-semantics).

GitOrigin-RevId: 3ca5d8082a0c6bd9520544ce3bca11bf3e02a5fa
4 files changed
tree: b7e602d30fa8c72be9ae8d82859c7ed31ac64b41
  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. README.md
README.md

Multi-Level Intermediate Representation

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