Skip to content

Commit 531a18e

Browse files
committed
add snake_case python binding
1 parent ba5aa77 commit 531a18e

File tree

2 files changed

+11
-1
lines changed

2 files changed

+11
-1
lines changed

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

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -158,3 +158,13 @@ def __init__(
158158
loc=loc,
159159
ip=ip,
160160
)
161+
162+
163+
def set_gpu_launch_threads(
164+
launch_op: Union[Operation, Value],
165+
threads: MixedValues,
166+
*,
167+
loc=None,
168+
ip=None,
169+
):
170+
return SetGPULaunchThreadsOp(launch_op, threads, loc=loc, ip=ip)

mlir/test/python/dialects/transform_xegpu_ext.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -123,7 +123,7 @@ def setGPULaunchThreadsOp():
123123
transform.OperationType.get("gpu.launch"),
124124
)
125125
with InsertionPoint(sequence.body):
126-
xegpu.SetGPULaunchThreadsOp(sequence.bodyTarget, threads=[8, 4, 1])
126+
xegpu.set_gpu_launch_threads(sequence.bodyTarget, threads=[8, 4, 1])
127127
transform.YieldOp()
128128
# CHECK-LABEL: TEST: setGPULaunchThreadsOp
129129
# CHECK: transform.xegpu.set_gpu_launch_threads

0 commit comments

Comments
 (0)