Skip to content

Commit 1a03d07

Browse files
committed
snake_case python functions returns op result value(s) if any
1 parent e9459dc commit 1a03d07

File tree

1 file changed

+4
-4
lines changed
  • mlir/python/mlir/dialects/transform

1 file changed

+4
-4
lines changed

mlir/python/mlir/dialects/transform/xegpu.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -47,8 +47,8 @@ def get_desc_op(
4747
*,
4848
loc=None,
4949
ip=None,
50-
) -> GetDescOp:
51-
return GetDescOp(target, loc=loc, ip=ip)
50+
) -> OpResult:
51+
return GetDescOp(target, loc=loc, ip=ip).result
5252

5353

5454
@_ods_cext.register_operation(_Dialect, replace=True)
@@ -105,15 +105,15 @@ def set_desc_layout(
105105
inst_data: Optional[MixedValues] = None,
106106
loc=None,
107107
ip=None,
108-
) -> SetDescLayoutOp:
108+
) -> OpResult:
109109
return SetDescLayoutOp(
110110
target,
111111
sg_layout,
112112
sg_data,
113113
inst_data=inst_data,
114114
loc=loc,
115115
ip=ip,
116-
)
116+
).result
117117

118118

119119
@_ods_cext.register_operation(_Dialect, replace=True)

0 commit comments

Comments
 (0)