[Mlir-commits] [mlir] [MLIR][XeGPU] add xegpu.set_desc_layout transform op (PR #165615)

[llvmlistbot at llvm.org]

github-actions[bot] wrote:

<!--LLVM CODE FORMAT COMMENT: {darker}-->


:warning: Python code formatter, darker found issues in your code. :warning:

<details>
<summary>
You can test this locally with the following command:
</summary>

``````````bash
darker --check --diff -r origin/main...HEAD mlir/python/mlir/dialects/transform/xegpu.py mlir/test/python/dialects/transform_xegpu_ext.py
``````````

:warning:
The reproduction instructions above might return results for more than one PR
in a stack if you are using a stacked PR workflow. You can limit the results by
changing `origin/main` to the base branch/commit you want to compare against.
:warning:

</details>

<details>
<summary>
View the diff from darker here.
</summary>

``````````diff
--- python/mlir/dialects/transform/xegpu.py	2025-10-29 19:23:23.000000 +0000
+++ python/mlir/dialects/transform/xegpu.py	2025-10-29 19:26:58.572453 +0000
@@ -59,7 +59,7 @@
             dynamic_inst_data,
             static_sg_layout=static_sg_layout,
             static_sg_data=static_sg_data,
             static_inst_data=static_inst_data,
             loc=loc,
-            ip=ip
+            ip=ip,
         )
--- test/python/dialects/transform_xegpu_ext.py	2025-10-29 19:23:23.000000 +0000
+++ test/python/dialects/transform_xegpu_ext.py	2025-10-29 19:26:58.590502 +0000
@@ -23,14 +23,11 @@
         [],
         transform.OperationType.get("xegpu.create_nd_tdesc"),
     )
     with InsertionPoint(sequence.body):
         xegpu.SetDescLayoutOp(
-            sequence.bodyTarget,
-            sg_layout=[6, 4],
-            sg_data=[32, 16],
-            inst_data=[8, 16]
+            sequence.bodyTarget, sg_layout=[6, 4], sg_data=[32, 16], inst_data=[8, 16]
         )
         transform.YieldOp()
     # CHECK-LABEL: TEST: setDescLayout
     # CHECK: %0 = transform.xegpu.set_desc_layout %
     # CHECK: sg_layout = [6, 4]
@@ -45,14 +42,11 @@
         [],
         transform.OperationType.get("xegpu.create_nd_tdesc"),
     )
     with InsertionPoint(sequence.body):
         xegpu.SetDescLayoutOp(
-            sequence.bodyTarget,
-            sg_layout=[6, 4],
-            sg_data=[32, 16],
-            inst_data=[8, 16]
+            sequence.bodyTarget, sg_layout=[6, 4], sg_data=[32, 16], inst_data=[8, 16]
         )
         transform.YieldOp()
     # CHECK-LABEL: TEST: setDescLayoutDefaultIndex
     # CHECK: %0 = transform.xegpu.set_desc_layout %
     # CHECK: sg_layout = [6, 4]

``````````

</details>


https://github.com/llvm/llvm-project/pull/165615