use darker to format python code.

linuxlonelyeagle · linuxlonelyeagle · commit f52d0c52f627 · 2025-04-13T19:14:40.000Z
diff --git a/mlir/test/Examples/NVGPU/tools/nvdsl.py b/mlir/test/Examples/NVGPU/tools/nvdsl.py
@@ -145,7 +145,10 @@ def create_descriptor(self, 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
 
diff --git a/mlir/test/Integration/GPU/CUDA/sm90/python/tools/matmulBuilder.py b/mlir/test/Integration/GPU/CUDA/sm90/python/tools/matmulBuilder.py
@@ -66,7 +66,10 @@ def tma_descriptor_op(self, 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
 

Original file line number	Diff line number	Diff line change
`@@ -145,7 +145,10 @@ def create_descriptor(self, device_ptr):`
`145`	`145`	`)`
`146`	`146`	`box_static_dim = [MLIR_DYNAMIC] * len(self.tma_box_shape)`
`147`	`147`	`self.tma_descriptor = nvgpu.TmaCreateDescriptorOp(`
`148`		`- tma_descriptor_ty, device_unranked_memref, map(const, self.tma_box_shape), box_static_dim`
	`148`	`+ tma_descriptor_ty,`
	`149`	`+ device_unranked_memref,`
	`150`	`+ map(const, self.tma_box_shape),`
	`151`	`+ box_static_dim,`
`149`	`152`	`)`
`150`	`153`	`return self.tma_descriptor.result`
`151`	`154`
Original file line number	Diff line number	Diff line change
`@@ -66,7 +66,10 @@ def tma_descriptor_op(self, device_ptr):`
`66`	`66`	`)`
`67`	`67`	`box_static_dim = [MLIR_DYNAMIC] * len(self.tma_box_shape)`
`68`	`68`	`tma_descriptor_op = nvgpu.TmaCreateDescriptorOp(`
`69`		`- tma_descriptor_ty, device_unranked_memref, map(c, self.tma_box_shape), box_static_dim`
	`69`	`+ tma_descriptor_ty,`
	`70`	`+ device_unranked_memref,`
	`71`	`+ map(c, self.tma_box_shape),`
	`72`	`+ box_static_dim,`
`70`	`73`	`)`
`71`	`74`	`return tma_descriptor_op.result`
`72`	`75`