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

[MLIR][XeGPU][TransformOps] Add set_gpu_launch_threads op (#166865)

Browse Source 
Adds `transform.xegpu.set_gpu_launch_threads` that overrides `gpu.launch` operation threads.
This commit is contained in:
Tuomas Kärnä
2025-11-11 13:57:54 +02:00
committed by GitHub
parent b440fb7584
commit 300750d4be
6 changed files with 263 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
mlir/test/python/dialects/transform_xegpu_ext.py
View File

@@ -113,3 +113,18 @@ def setOpLayoutAttrResult():
# CHECK: sg_layout = [6, 4]
# CHECK: sg_data = [32, 16]
# CHECK: inst_data = [8, 16]
@run
def setGPULaunchThreadsOp():
sequence = transform.SequenceOp(
transform.FailurePropagationMode.Propagate,
[],
transform.OperationType.get("gpu.launch"),
)
with InsertionPoint(sequence.body):
xegpu.set_gpu_launch_threads(sequence.bodyTarget, threads=[8, 4, 1])
transform.YieldOp()
# CHECK-LABEL: TEST: setGPULaunchThreadsOp
# CHECK: transform.xegpu.set_gpu_launch_threads
# CHECK: threads = [8, 4, 1]