[Mlir-commits] [mlir] [mlir][nvgpu] make TmaCreateDescriptorOp can use static box and add folder function to it. (PR #135497)

[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 HEAD~1...HEAD mlir/test/Examples/NVGPU/tools/nvdsl.py mlir/test/Integration/GPU/CUDA/sm90/python/tools/matmulBuilder.py
``````````

</details>

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

``````````diff
--- Examples/NVGPU/tools/nvdsl.py	2025-04-13 18:39:13.000000 +0000
+++ Examples/NVGPU/tools/nvdsl.py	2025-04-13 18:41:52.092769 +0000
@@ -143,11 +143,14 @@
             ),
             device_ptr,
         )
         box_static_dim = [MLIR_DYNAMIC] * len(self.tma_box_shape)
         self.tma_descriptor = nvgpu.TmaCreateDescriptorOp(
-            tma_descriptor_ty, device_unranked_memref, map(const, self.tma_box_shape), box_static_dim
+            tma_descriptor_ty,
+            device_unranked_memref,
+            map(const, self.tma_box_shape),
+            box_static_dim,
         )
         return self.tma_descriptor.result
 
     def prefetch(self, predicate=None):
         nvgpu.tma_prefetch_descriptor(self.tma_descriptor, predicate=predicate)
--- Integration/GPU/CUDA/sm90/python/tools/matmulBuilder.py	2025-04-13 18:39:13.000000 +0000
+++ Integration/GPU/CUDA/sm90/python/tools/matmulBuilder.py	2025-04-13 18:41:52.424695 +0000
@@ -64,11 +64,14 @@
             ),
             device_ptr,
         )
         box_static_dim = [MLIR_DYNAMIC] * len(self.tma_box_shape)
         tma_descriptor_op = nvgpu.TmaCreateDescriptorOp(
-            tma_descriptor_ty, device_unranked_memref, map(c, self.tma_box_shape), box_static_dim
+            tma_descriptor_ty,
+            device_unranked_memref,
+            map(c, self.tma_box_shape),
+            box_static_dim,
         )
         return tma_descriptor_op.result
 
 
 def debug_print(fmt, *args, predicate=None, threadNumber=-1, forcePrint=False):

``````````

</details>


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