[all-commits] [llvm/llvm-project] 78dc1e: [mlir][linalg][python] Add shape-only tensor suppo...

[Tobias Gysi via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 78dc1e497807e6b857fde7f78f4bc9cb5a4f8939
      https://github.com/llvm/llvm-project/commit/78dc1e497807e6b857fde7f78f4bc9cb5a4f8939
  Author: Tobias Gysi <gysit at google.com>
  Date:   2021-06-24 (Thu, 24 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/ops/core_named_ops.py
    M mlir/test/Dialect/Linalg/generalize-named-polymorphic-ops.mlir
    M mlir/test/python/dialects/linalg/opdsl/emit_structured_generic.py
    M mlir/test/python/dialects/linalg/opdsl/shape_maps_iteration.py
    M mlir/test/python/dialects/linalg/opsrun.py

  Log Message:
  -----------
  [mlir][linalg][python] Add shape-only tensor support to OpDSL.

Add an index_dim annotation to specify the shape to loop mapping of shape-only tensors. A shape-only tensor serves is not accessed withing the body of the operation but is required to span the iteration space of certain operations such as pooling.

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