[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