[all-commits] [llvm/llvm-project] ea4a51: [mlir][linalg][transform][python] Refactor TileOp ...

[Ingo Müller via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: ea4a5127c49d01e7415054e9ebeb813c98159810
      https://github.com/llvm/llvm-project/commit/ea4a5127c49d01e7415054e9ebeb813c98159810
  Author: Ingo Müller <ingomueller at google.com>
  Date:   2023-09-04 (Mon, 04 Sep 2023)

  Changed paths:
    M mlir/python/mlir/dialects/_structured_transform_ops_ext.py
    M mlir/test/python/dialects/transform_structured_ext.py

  Log Message:
  -----------
  [mlir][linalg][transform][python] Refactor TileOp mix-in.

This patch simplifies and improves the mix-in of the `TileOp`. In
particular:

* Accept all types of sizes (static, dynamic, scalable) in a single
  argument `sizes`.
* Use the existing convenience function to dispatch different types of
  sizes instead of repeating the implementation in the mix-in.
* Pass on `None` values as is of optional arguments to the init function
  of the super class.
* Reformat with default indentation width (4 spaces vs 2 spaces).
* Add a a test for providing scalable sizes.

Reviewed By: ftynse

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