Address renaming feedback

deepnvme/file_access/aio_load_cpu_tensor.py

@@ -3,8 +3,8 @@
 from deepspeed.ops.op_builder import AsyncIOBuilder
 from utils import parse_read_arguments, GIGA_UNIT
 
-def file_read(inp_f, h, bounce_buffer):
-    h.sync_pread(bounce_buffer, inp_f)
+def file_read(inp_f, handle, bounce_buffer):
+    handle.sync_pread(bounce_buffer, inp_f)
     return bounce_buffer.cpu()
 
 def main():

deepnvme/file_access/aio_load_gpu_tensor.py

@@ -3,8 +3,8 @@
 from deepspeed.ops.op_builder import AsyncIOBuilder
 from utils import parse_read_arguments, GIGA_UNIT
 
-def file_read(inp_f, h, bounce_buffer):
-    h.sync_pread(bounce_buffer, inp_f)
+def file_read(inp_f, handle, bounce_buffer):
+    handle.sync_pread(bounce_buffer, inp_f)
     return bounce_buffer.cuda()
 
 

deepnvme/file_access/aio_store_cpu_tensor.py

@@ -3,9 +3,9 @@
 from deepspeed.ops.op_builder import AsyncIOBuilder
 from utils import parse_write_arguments, GIGA_UNIT
 
-def file_write(out_f, t, h, bounce_buffer):
-    bounce_buffer.copy_(t)
-    h.sync_pwrite(bounce_buffer, out_f)
+def file_write(out_f, tensor, handle, bounce_buffer):
+    bounce_buffer.copy_(tensor)
+    handle.sync_pwrite(bounce_buffer, out_f)
 
 def main():
     args = parse_write_arguments()

deepnvme/file_access/aio_store_gpu_tensor.py

@@ -3,9 +3,9 @@
 from deepspeed.ops.op_builder import AsyncIOBuilder
 from utils import parse_write_arguments, GIGA_UNIT
 
-def file_write(out_f, t, h, bounce_buffer):
-    bounce_buffer.copy_(t)
-    h.sync_pwrite(bounce_buffer, out_f)
+def file_write(out_f, tensor, handle, bounce_buffer):
+    bounce_buffer.copy_(tensor)
+    handle.sync_pwrite(bounce_buffer, out_f)
 
 def main():
     args = parse_write_arguments()

deepnvme/file_access/gds_load_gpu_tensor.py

@@ -3,8 +3,8 @@
 from utils import parse_read_arguments, GIGA_UNIT
 from deepspeed.ops.op_builder import GDSBuilder
 
-def file_read(inp_f, h, gpu_buffer):
-    h.sync_pread(gpu_buffer, inp_f)
+def file_read(inp_f, handle, gpu_buffer):
+    handle.sync_pread(gpu_buffer, inp_f)
     return gpu_buffer.cuda()
 
 def main():

deepnvme/file_access/gds_store_gpu_tensor.py

@@ -3,9 +3,9 @@
 from deepspeed.ops.op_builder import GDSBuilder
 from utils import parse_write_arguments, GIGA_UNIT
 
-def file_write(out_f, t, h, gpu_buffer):
-    gpu_buffer.copy_(t)
-    h.sync_pwrite(gpu_buffer, out_f)
+def file_write(out_f, tensor, handle, gpu_buffer):
+    gpu_buffer.copy_(tensor)
+    handle.sync_pwrite(gpu_buffer, out_f)
 
 def main():
     args = parse_write_arguments()

deepnvme/file_access/py_load_cpu_tensor.py

@@ -4,8 +4,8 @@
 
 def file_read(inp_f):
     with open(inp_f, 'rb') as f:
-       t = torch.frombuffer(f.read(), dtype=torch.uint8)
-    return t 
+       tensor = torch.frombuffer(f.read(), dtype=torch.uint8)
+    return tensor
 
 def main():
     args = parse_read_arguments()

deepnvme/file_access/py_load_gpu_tensor.py

@@ -4,8 +4,8 @@
 
 def file_read(inp_f):
     with open(inp_f, 'rb') as f:
-       t = torch.frombuffer(f.read(), dtype=torch.uint8)
-    return t.cuda()
+       tensor = torch.frombuffer(f.read(), dtype=torch.uint8)
+    return tensor.cuda()
 
 def main():
     args = parse_read_arguments()

deepnvme/file_access/py_store_cpu_tensor.py

@@ -3,9 +3,9 @@
 import pathlib
 from utils import parse_write_arguments, GIGA_UNIT
 
-def file_write(out_f, t):
+def file_write(out_f, tensor):
     with open(out_f, 'wb') as f:
-       f.write(t.numpy(force=True))
+       f.write(tensor.numpy(force=True))
 
 def main():
     args = parse_write_arguments()

deepnvme/file_access/py_store_gpu_tensor.py

@@ -3,9 +3,9 @@
 import pathlib
 from utils import parse_write_arguments, GIGA_UNIT
 
-def file_write(out_f, t):
+def file_write(out_f, tensor):
     with open(out_f, 'wb') as f:
-       f.write(t.numpy(force=True))
+       f.write(tensor.numpy(force=True))
 
 def main():
     args = parse_write_arguments()