[all-commits] [llvm/llvm-project] 3b9540: [mlir][linalg][python] Add max operation in OpDSL

Tobias Gysi via All-commits all-commits at lists.llvm.org
Fri Jul 2 00:13:12 PDT 2021


  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 3b95400f78a9824172629123580c0a0df36cbc70
      https://github.com/llvm/llvm-project/commit/3b95400f78a9824172629123580c0a0df36cbc70
  Author: Tobias Gysi <gysit at google.com>
  Date:   2021-07-02 (Fri, 02 Jul 2021)

  Changed paths:
    M mlir/include/mlir/Dialect/Linalg/IR/LinalgNamedStructuredOps.yaml
    M mlir/lib/Dialect/Linalg/IR/LinalgOps.cpp
    M mlir/python/mlir/dialects/linalg/opdsl/lang/emitter.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/opsrun.py

  Log Message:
  -----------
  [mlir][linalg][python] Add max operation in OpDSL

Add the max operation to the OpDSL and introduce a max pooling operation to test the implementation. As MLIR has no builtin max operation, the max function is lowered to a compare and select pair.

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




More information about the All-commits mailing list