intel/llvm
1
0
Fork 0
mirror of https://github.com/intel/llvm.git synced 2026-02-02 02:00:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

Revert "[mlir][linalg][transform][python] Add mix-in for MapCopyToThreadsOp."

Browse Source 
This reverts commit 691a2fab88.

The bot is broken: https://lab.llvm.org/buildbot/#/builders/61/builds/47577
This commit is contained in:
Mehdi Amini
2023-08-14 08:56:24 -07:00
parent 9184eb84a1
commit ecc4ef9f2b
2 changed files with 0 additions and 98 deletions
Download Patch File Download Diff File Expand all files Collapse all files

60
mlir/python/mlir/dialects/_structured_transform_ops_ext.py
View File

@@ -221,66 +221,6 @@ class InterchangeOp:
)
class MapCopyToThreadsOp:
"""Specialization for MapCopyToThreadsOp class."""
@overload
def __init__(
self,
forall_op_type: Type,
tiled_op_type: Type,
target: Union[Operation, OpView, Value],
*,
total_num_threads: Union[int, IntegerAttr],
desired_bit_alignment: Union[int, IntegerAttr],
loc=None,
ip=None,
):
...
@overload
def __init__(
self,
target: Union[Operation, OpView, Value],
*,
total_num_threads: Union[int, IntegerAttr],
desired_bit_alignment: Union[int, IntegerAttr],
loc=None,
ip=None,
):
...
def __init__(
self,
forall_op_type_or_target: Union[Operation, OpView, Type, Value],
tiled_op_type_or_none: Optional[Type] = None,
target_or_none: Optional[Union[Operation, OpView, Value]] = None,
*,
total_num_threads: Union[int, IntegerAttr],
desired_bit_alignment: Union[int, IntegerAttr],
loc=None,
ip=None,
):
if isinstance(forall_op_type_or_target, Type):
forall_op_type = forall_op_type_or_target
tiled_op_type = tiled_op_type_or_none
target = target_or_none
else:
forall_op_type = transform.AnyOpType.get()
tiled_op_type = transform.AnyOpType.get()
target = forall_op_type_or_target
super().__init__(
forall_op_type,
tiled_op_type,
target,
total_num_threads=total_num_threads,
desired_bit_alignment=desired_bit_alignment,
loc=loc,
ip=ip,
)
class MatchOp:
"""Specialization for MatchOp class."""