rename into memsys stats enabled

Author: isVoid

numba_cuda/numba/cuda/dispatcher.py

@@ -365,7 +365,7 @@ def launch(self, args, griddim, blockdim, stream=0, sharedmem=0):
         rtsys.ensure_allocate(stream_handle)
         rtsys.set_memsys_to_module(cufunc.module, stream_handle)
         rtsys.ensure_initialize(stream_handle)
-        rtsys.enable(stream_handle)
+        rtsys.memsys_stats_enabled(stream_handle)
 
         # Invoke kernel
         driver.launch_kernel(cufunc.handle,

numba_cuda/numba/cuda/runtime/nrt.py

@@ -84,11 +84,11 @@ def initialize(self, stream):
             self._memsys_module, stream, "NRT_MemSys_init")
         self._initialized = True
 
-    def enable(self, stream):
+    def memsys_stats_enabled(self, stream):
         self._single_thread_launch(
             self._memsys_module, stream, "NRT_MemSys_enable")
 
-    def disable(self, stream):
+    def memsys_stats_disabled(self, stream):
         self._single_thread_launch(
             self._memsys_module, stream, "NRT_MemSys_disable")
 
@@ -116,8 +116,8 @@ def _copy_memsys_to_host(self, stream=0):
 
         return stats_for_read[0]
 
-    def get_allocation_stats(self):
-        memsys = self._copy_memsys_to_host()
+    def get_allocation_stats(self, stream=0):
+        memsys = self._copy_memsys_to_host(stream)
         return _nrt_mstats(
             alloc=memsys["alloc"],
             free=memsys["free"],