[Mlir-commits] [mlir] [MLIR, Python] Support converting boolean numpy arrays to and from mlir attributes (PR #113064)

[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 9120adea504981dfd55ace25825f84018543d6f2...52b49ac835f82c9d5ca9c1ae8f433e0a685e23a5 mlir/test/python/ir/array_attributes.py
``````````

</details>

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

``````````diff
--- array_attributes.py	2024-10-19 20:00:56.000000 +0000
+++ array_attributes.py	2024-10-19 20:46:17.599829 +0000
@@ -344,39 +344,47 @@
         print(attr)
         # CHECK: {{\[}}[ True False True]
         # CHECK: {{\[}} True True False]]
         print(np.array(attr))
 
-        array = np.array([[True, True, False, False], [True, False, True, False]], dtype=np.bool_)
+        array = np.array(
+            [[True, True, False, False], [True, False, True, False]], dtype=np.bool_
+        )
         attr = DenseElementsAttr.get(array)
         # CHECK: dense<{{\[}}[true, true, false, false], [true, false, true, false]]> : tensor<2x4xi1>
         print(attr)
         # CHECK: {{\[}}[ True True False False]
         # CHECK: {{\[}} True False True False]]
         print(np.array(attr))
 
-        array = np.array([
-            [True, True, False, False],
-            [True, False, True, False],
-            [False, False, False, False],
-            [True, True, True, True],
-            [True, False, False, True],
-        ], dtype=np.bool_)
+        array = np.array(
+            [
+                [True, True, False, False],
+                [True, False, True, False],
+                [False, False, False, False],
+                [True, True, True, True],
+                [True, False, False, True],
+            ],
+            dtype=np.bool_,
+        )
         attr = DenseElementsAttr.get(array)
         # CHECK: dense<{{\[}}[true, true, false, false], [true, false, true, false], [false, false, false, false], [true, true, true, true], [true, false, false, true]]> : tensor<5x4xi1>
         print(attr)
         # CHECK: {{\[}}[ True True False False]
         # CHECK: {{\[}} True False True False]
         # CHECK: {{\[}}False False False False]
         # CHECK: {{\[}} True True True True]
         # CHECK: {{\[}} True False False True]]
         print(np.array(attr))
 
-        array = np.array([
-            [True, True, False, False, True, True, False, False, False],
-            [False, False, False, True, False, True, True, False, True]
-        ], dtype=np.bool_)
+        array = np.array(
+            [
+                [True, True, False, False, True, True, False, False, False],
+                [False, False, False, True, False, True, True, False, True],
+            ],
+            dtype=np.bool_,
+        )
         attr = DenseElementsAttr.get(array)
         # CHECK: dense<{{\[}}[true, true, false, false, true, true, false, false, false], [false, false, false, true, false, true, true, false, true]]> : tensor<2x9xi1>
         print(attr)
         # CHECK: {{\[}}[ True True False False True True False False False]
         # CHECK: {{\[}}False False False True False True True False True]]

``````````

</details>


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