update api name

Author: mag1c-h

ucm/store/cache/connector.py

@@ -105,7 +105,7 @@ def dump(
         task_id = self.store.Dump(ids, indexes, addrs)
         return CacheTransTask(task_id)
 
-    def fetch_data(
+    def load_data(
         self,
         block_ids: List[bytes],
         shard_index: List[int],

ucm/store/posix/connector.py

@@ -104,7 +104,7 @@ def dump(
         task_id = self.store.Dump(ids, indexes, addrs)
         return PosixTransTask(task_id)
 
-    def fetch_data(
+    def load_data(
         self,
         block_ids: List[bytes],
         shard_index: List[int],
@@ -162,5 +162,5 @@ def check(self, task: Task) -> bool:
     founds = store.lookup(block_ids)
     assert all(founds)
     data2 = [[cupy.cuda.alloc_pinned_memory(block_size).ptr] for _ in range(block_num)]
-    handle = store.fetch_data(block_ids, shard_idxes, data2)
+    handle = store.load_data(block_ids, shard_idxes, data2)
     store.wait(handle)

ucm/store/test/e2e/cache_on_posix_test.py

@@ -22,10 +22,10 @@
 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 # SOFTWARE.
 #
-from abc import ABC
 import os
 import secrets
 import time
+from abc import ABC
 from typing import List
 
 import cupy
@@ -88,13 +88,13 @@ def dump(
     ) -> Task:
         return self.cache.dump(block_ids, shard_index, src_tensor)
 
-    def fetch_data(
+    def load_data(
         self,
         block_ids: List[bytes],
         shard_index: List[int],
         dst_addr: List[List[int]],
     ) -> Task:
-        return self.cache.fetch_data(block_ids, shard_index, dst_addr)
+        return self.cache.load_data(block_ids, shard_index, dst_addr)
 
     def dump_data(
         self,
@@ -141,7 +141,7 @@ def main():
     ]
     founds = store.lookup(chunk_block_ids)
     assert all(founds)
-    task = store.fetch_data(chunk_block_ids, shard_indexes, dst_addrs)
+    task = store.load_data(chunk_block_ids, shard_indexes, dst_addrs)
     store.wait(task)
 
 

ucm/store/ucmstore.py

@@ -126,7 +126,7 @@ def dump(
         pass
 
     @abstractmethod
-    def fetch_data(
+    def load_data(
         self,
         block_ids: List[bytes],
         shard_index: List[int],
@@ -169,7 +169,7 @@ def wait(self, task: Task) -> None:
         """Block until the given transfer task completes.
 
         Args:
-            task: Task handle returned by ``load``, ``dump``, ``fetch_data``,
+            task: Task handle returned by ``load``, ``dump``, ``load_data``,
                 or ``dump_data``.
         """
         pass