Skip to content

Commit 04a4821

Browse files
committed
schedule: add comment about gpu-lower-to-xevm-pipeline
1 parent ee34b37 commit 04a4821

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

python/examples/xegpu_matmul/schedule.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -370,6 +370,11 @@ def bundle_xegpu_to_binary(mod, dump_kernel: str = ""):
370370
# This schedule corresponds to upstream MLIR XeVM lowering pipeline
371371
# and is payload independent.
372372

373+
# TODO applying gpu-lower-to-xevm-pipeline pass affects performance
374+
# mod = apply_registered_pass(
375+
# mod, "gpu-lower-to-xevm-pipeline", options={"xegpu-op-level": "workgroup"}
376+
# )
377+
373378
gpu_mod = match(mod, ops={"gpu.module"})
374379
# xegpu distribution
375380
gpu_func = match(gpu_mod, ops={"gpu.func"})

0 commit comments

Comments
 (0)