Remove deprecation warning in 4.5 (Dao-AILab#139)

drisspg · web-flow · commit c88309ba6759 · 2026-05-17T00:20:16.000-04:00
diff --git a/quack/gemm_sm100.py b/quack/gemm_sm100.py
@@ -968,11 +968,11 @@ def kernel(
         )
         # Tensor memory dealloc barrier init
         tmem = cutlass.utils.TmemAllocator(
-            storage.tmem_holding_buf,
+            storage.tmem_holding_buf.ptr,
             barrier_for_retrieve=tmem_alloc_barrier,
             allocator_warp_id=self.epilog_warp_id[0],
             is_two_cta=use_2cta_instrs,
-            two_cta_tmem_dealloc_mbar_ptr=storage.tmem_dealloc_mbar_ptr,
+            two_cta_tmem_dealloc_mbar_ptr=storage.tmem_dealloc_mbar_ptr.ptr,
         )
 
         # Cluster arrive after barrier init

Original file line number	Diff line number	Diff line change
`@@ -968,11 +968,11 @@ def kernel(`
`968`	`968`	`)`
`969`	`969`	`# Tensor memory dealloc barrier init`
`970`	`970`	`tmem = cutlass.utils.TmemAllocator(`
`971`		`- storage.tmem_holding_buf,`
	`971`	`+ storage.tmem_holding_buf.ptr,`
`972`	`972`	`barrier_for_retrieve=tmem_alloc_barrier,`
`973`	`973`	`allocator_warp_id=self.epilog_warp_id[0],`
`974`	`974`	`is_two_cta=use_2cta_instrs,`
`975`		`- two_cta_tmem_dealloc_mbar_ptr=storage.tmem_dealloc_mbar_ptr,`
	`975`	`+ two_cta_tmem_dealloc_mbar_ptr=storage.tmem_dealloc_mbar_ptr.ptr,`
`976`	`976`	`)`
`977`	`977`
`978`	`978`	`# Cluster arrive after barrier init`