[all-commits] [llvm/llvm-project] f345f7: [mlir][OpDSL] Support pointwise ops with rank zero...

[Tobias Gysi via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: f345f7e30bd3a8e15052f5669c1977aa088e468f
      https://github.com/llvm/llvm-project/commit/f345f7e30bd3a8e15052f5669c1977aa088e468f
  Author: gysit <gysit at google.com>
  Date:   2022-03-08 (Tue, 08 Mar 2022)

  Changed paths:
    M mlir/python/mlir/dialects/linalg/opdsl/lang/emitter.py
    M mlir/test/Dialect/Linalg/generalize-named-polymorphic-ops.mlir
    M mlir/test/python/dialects/linalg/opdsl/emit_fill.py
    M mlir/test/python/integration/dialects/linalg/opsrun.py
    M mlir/tools/mlir-linalg-ods-gen/mlir-linalg-ods-yaml-gen.cpp

  Log Message:
  -----------
  [mlir][OpDSL] Support pointwise ops with rank zero inputs.

Allow pointwise operations to take rank zero input tensors similarly to scalar inputs. Use an empty indexing map to broadcast rank zero tensors to the iteration domain of the operation.

Depends On D120734

Reviewed By: nicolasvasilache

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