[Mlir-commits] [mlir] [mlir][Python] fix linalg and python_test dialect extensions (PR #175405)

[llvmlistbot at llvm.org]

github-actions[bot] wrote:

<!--PREMERGE ADVISOR COMMENT: Linux-->
# :penguin: Linux x64 Test Results

* 7317 tests passed
* 598 tests skipped
* 1 test failed

## Failed Tests
(click on a test name to see its output)

### MLIR
<details>
<summary>MLIR.python/dialects/linalg/utils.py</summary>

```
Exit Code: 1

Command Output (stdout):
--
# RUN: at line 1
/usr/bin/python3 /home/gha/actions-runner/_work/llvm-project/llvm-project/mlir/test/python/dialects/linalg/utils.py
# executed command: /usr/bin/python3 /home/gha/actions-runner/_work/llvm-project/llvm-project/mlir/test/python/dialects/linalg/utils.py
# .---command stdout------------
# | 
# | TEST: test_infer_contraction_dimensions_from_ops
# `-----------------------------
# .---command stderr------------
# | Traceback (most recent call last):
# |   File "/home/gha/actions-runner/_work/llvm-project/llvm-project/mlir/test/python/dialects/linalg/utils.py", line 14, in <module>
# |     @run
# |      ^^^
# |   File "/home/gha/actions-runner/_work/llvm-project/llvm-project/mlir/test/python/dialects/linalg/utils.py", line 10, in run
# |     f()
# |   File "/home/gha/actions-runner/_work/llvm-project/llvm-project/mlir/test/python/dialects/linalg/utils.py", line 26, in test_infer_contraction_dimensions_from_ops
# |     @func.FuncOp.from_py_func(a_type, b_type, c_type)
# |      ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
# |   File "/home/gha/actions-runner/_work/llvm-project/llvm-project/build/tools/mlir/python_packages/mlir_core/mlir/dialects/func.py", line 197, in decorator
# |     return_values = f(*func_args, **func_kwargs)
# |                     ^^^^^^^^^^^^^^^^^^^^^^^^^^^^
# |   File "/home/gha/actions-runner/_work/llvm-project/llvm-project/mlir/test/python/dialects/linalg/utils.py", line 57, in contraction_fn
# |     assert list(dims.n) == [1], f"Expected n=[1], got {list(dims.n)}"
# |            ^^^^^^^^^^^^^^^^^^^
# | AssertionError: Expected n=[1], got [2]
# `-----------------------------
# error: command failed with exit status: 1

--

```
</details>

If these failures are unrelated to your changes (for example tests are broken or flaky at HEAD), please open an issue at https://github.com/llvm/llvm-project/issues and add the `infrastructure` label.

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