[Mlir-commits] [mlir] [MLIR] Extend linalg.pack and linalg.unpack to accept memref (PR #167675)

[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/linalg/__init__.py mlir/test/python/dialects/linalg/ops.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
--- test/python/dialects/linalg/ops.py	2026-01-16 17:08:12.000000 +0000
+++ test/python/dialects/linalg/ops.py	2026-01-16 17:09:56.737970 +0000
@@ -667,16 +667,11 @@
             @func.FuncOp.from_py_func(
                 MemRefType.get((128, 128), f32),
                 MemRefType.get((16, 16, 8, 8), f32),
             )
             def memref_pack(src, dst):
-                linalg.pack(
-                    src,
-                    dst,
-                    inner_dims_pos=[1, 0],
-                    inner_tiles=[8, 8]
-                )
+                linalg.pack(src, dst, inner_dims_pos=[1, 0], inner_tiles=[8, 8])
 
                 linalg.unpack(
                     dst,
                     src,
                     inner_dims_pos=[0, 1],

``````````

</details>


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