Added test for the compiler memory refresh.

Author: isazi

kernel_tuner/backends/compiler.py

@@ -364,7 +364,7 @@ def memcpy_htod(self, dest, src):
         pass
 
     def refresh_memory(self, arguments, should_sync):
-        """Copy the preserved content of the output memory to device pointers."""
+        """Copy the preserved content of the output memory to used arrays."""
         for i, arg in enumerate(arguments):
             if should_sync[i]:
                 if isinstance(arg, np.ndarray) and is_cupy_array(self.allocations[i].numpy):

test/test_compiler_functions.py

@@ -368,3 +368,14 @@ def test_run_kernel():
         lang="C",
     )
     assert cp.all((a + b) == c)
+
+
+def test_refresh_memory():
+    arg1 = np.array([1, 2, 3]).astype(np.int8)
+    cfunc = CompilerFunctions()
+    output = cfunc.ready_argument_list([arg1])
+    assert np.all(output == arg1)
+    arg1 = np.array([0, 0, 0]).astype(np.int8)
+    assert np.all(arg1 == [0, 0, 0])
+    cfunc.refresh_memory(arg1, [True])
+    assert np.all(arg1 == [1, 2, 3])