[all-commits] [llvm/llvm-project] be6e9d: [mlir][transform][linalg][python] Add extended Til...

Ingo Müller via All-commits all-commits at lists.llvm.org
Wed Jul 19 07:02:44 PDT 2023


  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: be6e9df11f880ab128aef6550c6911d9f091e7d7
      https://github.com/llvm/llvm-project/commit/be6e9df11f880ab128aef6550c6911d9f091e7d7
  Author: Ingo Müller <ingomueller at google.com>
  Date:   2023-07-19 (Wed, 19 Jul 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][transform][linalg][python] Add extended TileToForallOp.

This patch adds a mixin for TileToForallOp to
_structured_transform_ops_ext.py with syntactic sugar for construction
such ops. First, the types of the results are made optional and filled
with common default values if omitted. Second, for num_threads and
tile_sizes, the three possible forms (static, dynamic, or packed), can
now all be given through the same respective argument, which gets
dispatched to the correct form-specific argument automatically.

Reviewed By: nicolasvasilache, ftynse

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




More information about the All-commits mailing list