Fix rmm deallocate arguments

gartavanis · Dec 12, 2024 · 5821131 · 5821131
1 parent 1b3a6f2
commit 5821131
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/pfsimulator/amps/common/rmm_wrapper.cpp b/pfsimulator/amps/common/rmm_wrapper.cpp
@@ -15,12 +15,12 @@ extern "C" {
 
   void* rmmAlloc(size_t bytes) {
     rmm::mr::device_memory_resource* mr = rmm::mr::get_current_device_resource(); // Points to `pool_mr`
-    return mr.allocate(bytes, cuda_stream_view(0));
+    return mr.allocate(bytes);
   }
 
-  void rmmFree(void *p, size_t bytes) {
+  void rmmFree(void *p) {
     rmm::mr::device_memory_resource* mr = rmm::mr::get_current_device_resource(); // Points to `pool_mr`
-    mr.deallocate(p, bytes, cuda_stream_view(0));
+    mr.deallocate(p);
   }
 }
 

diff --git a/pfsimulator/amps/common/rmm_wrapper.h b/pfsimulator/amps/common/rmm_wrapper.h
@@ -10,7 +10,7 @@ extern "C" {
 
 void rmmInit();
 void* rmmAlloc(size_t bytes);
-void rmmFree(void *p, size_t bytes);
+void rmmFree(void *p);
 
 #ifdef __cplusplus
 }