[mlir][transform][gpu][python] Add MapForallToBlocks mix-in.
authorIngo Müller <ingomueller@google.com>
Wed, 19 Jul 2023 15:31:23 +0000 (15:31 +0000)
committerIngo Müller <ingomueller@google.com>
Thu, 20 Jul 2023 14:20:40 +0000 (14:20 +0000)
commit5f4f9220f9d2f0163b330e4a2e2a60ea8a917378
treee72b6bef3b12925c381e9502721d96e193891aaa
parent53c936a67c15bc5025ef9f7804399884ca73efa7
[mlir][transform][gpu][python] Add MapForallToBlocks mix-in.

This patch adds a mix-in class for MapForallToBlocks with overloaded
constructors. This makes it optional to provide the return type of the
op, which is defaulte to `AnyOpType`.

Reviewed By: ftynse

Differential Revision: https://reviews.llvm.org/D155717
mlir/python/CMakeLists.txt
mlir/python/mlir/dialects/_gpu_transform_ops_ext.py [new file with mode: 0644]
mlir/test/python/dialects/transform_gpu_ext.py [new file with mode: 0644]