fix create_descriptor.

linuxlonelyeagle · linuxlonelyeagle · commit 891c30d8ec1b · 2025-04-13T18:39:13.000Z
diff --git a/mlir/test/Examples/NVGPU/tools/nvdsl.py b/mlir/test/Examples/NVGPU/tools/nvdsl.py
@@ -143,8 +143,9 @@ def create_descriptor(self, device_ptr):
             ),
             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)
+            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
@@ -64,8 +64,9 @@ def tma_descriptor_op(self, device_ptr):
             ),
             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)
+            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
`@@ -143,8 +143,9 @@ def create_descriptor(self, device_ptr):`
`143`	`143`	`),`
`144`	`144`	`device_ptr,`
`145`	`145`	`)`
	`146`	`+ box_static_dim = [MLIR_DYNAMIC] * len(self.tma_box_shape)`
`146`	`147`	`self.tma_descriptor = nvgpu.TmaCreateDescriptorOp(`
`147`		`- tma_descriptor_ty, device_unranked_memref, map(const, self.tma_box_shape)`
	`148`	`+ tma_descriptor_ty, device_unranked_memref, map(const, self.tma_box_shape), box_static_dim`
`148`	`149`	`)`
`149`	`150`	`return self.tma_descriptor.result`
`150`	`151`
Original file line number	Diff line number	Diff line change
`@@ -64,8 +64,9 @@ def tma_descriptor_op(self, device_ptr):`
`64`	`64`	`),`
`65`	`65`	`device_ptr,`
`66`	`66`	`)`
	`67`	`+ box_static_dim = [MLIR_DYNAMIC] * len(self.tma_box_shape)`
`67`	`68`	`tma_descriptor_op = nvgpu.TmaCreateDescriptorOp(`
`68`		`- tma_descriptor_ty, device_unranked_memref, map(c, self.tma_box_shape)`
	`69`	`+ tma_descriptor_ty, device_unranked_memref, map(c, self.tma_box_shape), box_static_dim`
`69`	`70`	`)`
`70`	`71`	`return tma_descriptor_op.result`
`71`	`72`