Skip to content

Commit ee597cd

Browse files
committed
snake_case python functions returns op result value(s) if any
1 parent 280693d commit ee597cd

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
@@ -46,8 +46,8 @@ def get_desc_op(
4646
*,
4747
loc=None,
4848
ip=None,
49-
) -> GetDescOp:
50-
return GetDescOp(target, loc=loc, ip=ip)
49+
) -> OpResult:
50+
return GetDescOp(target, loc=loc, ip=ip).result
5151

5252

5353
@_ods_cext.register_operation(_Dialect, replace=True)
@@ -104,15 +104,15 @@ def set_desc_layout(
104104
inst_data: Optional[MixedValues] = None,
105105
loc=None,
106106
ip=None,
107-
) -> SetDescLayoutOp:
107+
) -> OpResult:
108108
return SetDescLayoutOp(
109109
target,
110110
sg_layout,
111111
sg_data,
112112
inst_data=inst_data,
113113
loc=loc,
114114
ip=ip,
115-
)
115+
).result
116116

117117

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

0 commit comments

Comments
 (0)