Fix missing deallocate in cub

pciolkosz · pciolkosz · commit 861b9345f245 · 2026-03-20T09:57:34.000-07:00
diff --git a/cub/cub/detail/device_memory_resource.cuh b/cub/cub/detail/device_memory_resource.cuh
@@ -68,6 +68,11 @@ struct device_memory_resource
     return ptr;
   }
 
+  CUB_RUNTIME_FUNCTION void deallocate(::cuda::stream_ref stream, void* ptr, size_t bytes, size_t /* alignment */)
+  {
+    deallocate(ptr, bytes);
+  }
+
   CUB_RUNTIME_FUNCTION void deallocate(::cuda::stream_ref stream, void* ptr, size_t /* bytes */)
   {
     NV_IF_TARGET( //

Original file line number	Diff line number	Diff line change
`@@ -68,6 +68,11 @@ struct device_memory_resource`
`68`	`68`	`return ptr;`
`69`	`69`	`}`
`70`	`70`
	`71`	`+ CUB_RUNTIME_FUNCTION void deallocate(::cuda::stream_ref stream, void* ptr, size_t bytes, size_t /* alignment */)`
	`72`	`+ {`
	`73`	`+ deallocate(ptr, bytes);`
	`74`	`+ }`
	`75`	`+`
`71`	`76`	`CUB_RUNTIME_FUNCTION void deallocate(::cuda::stream_ref stream, void* ptr, size_t /* bytes */)`
`72`	`77`	`{`
`73`	`78`	`NV_IF_TARGET( //`