[all-commits] [llvm/llvm-project] 4361bd: [mlir][linalg][python] Explicit shape and dimensio...

[Tobias Gysi via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 4361bd9b7b38c73b69f9a37e52d0b72989e84947
      https://github.com/llvm/llvm-project/commit/4361bd9b7b38c73b69f9a37e52d0b72989e84947
  Author: Tobias Gysi <gysit at google.com>
  Date:   2021-06-30 (Wed, 30 Jun 2021)

  Changed paths:
    M mlir/include/mlir/Dialect/Linalg/IR/LinalgNamedStructuredOps.yaml
    M mlir/python/mlir/dialects/linalg/opdsl/lang/comprehension.py
    M mlir/python/mlir/dialects/linalg/opdsl/lang/config.py
    M mlir/python/mlir/dialects/linalg/opdsl/lang/dsl.py
    M mlir/python/mlir/dialects/linalg/opdsl/ops/core_named_ops.py
    M mlir/test/python/dialects/linalg/opdsl/arguments.py
    M mlir/test/python/dialects/linalg/opdsl/emit_structured_generic.py
    M mlir/test/python/dialects/linalg/opdsl/interfaces.py
    M mlir/test/python/dialects/linalg/opdsl/shape_maps_iteration.py

  Log Message:
  -----------
  [mlir][linalg][python] Explicit shape and dimension order in OpDSL.

Extend the OpDSL syntax with an optional `domain` function to specify an explicit dimension order. The extension is needed to provide more control over the dimension order instead of deducing it implicitly depending on the formulation of the tensor comprehension. Additionally, the patch also ensures the symbols are ordered according to the operand definitions of the operation.

Differential Revision: https://reviews.llvm.org/D105117