Handle init with default args

deepspeedai · Aug 16, 2024 · a595e39 · a595e39
1 parent 9b68819
commit a595e39
Show file tree

Hide file tree

Showing 6 changed files with 6 additions and 6 deletions.
diff --git a/deepnvme/file_access/aio_load_cpu_tensor.py b/deepnvme/file_access/aio_load_cpu_tensor.py
@@ -13,7 +13,7 @@ def main():
     file_sz = os.path.getsize(input_file)
     cnt = args.loop
 
-    aio_handle = AsyncIOBuilder().load().aio_handle(1024**2, 128, True, True, 1)
+    aio_handle = AsyncIOBuilder().load().aio_handle()
     bounce_buffer = torch.empty(os.path.getsize(input_file), dtype=torch.uint8).pin_memory()
 
     t = timeit.Timer(functools.partial(file_read, input_file, aio_handle, bounce_buffer))

diff --git a/deepnvme/file_access/aio_load_gpu_tensor.py b/deepnvme/file_access/aio_load_gpu_tensor.py
@@ -14,7 +14,7 @@ def main():
     file_sz = os.path.getsize(input_file)
     cnt = args.loop
 
-    aio_handle = AsyncIOBuilder().load().aio_handle(1024**2, 128, True, True, 1)
+    aio_handle = AsyncIOBuilder().load().aio_handle()
     bounce_buffer = torch.empty(os.path.getsize(input_file), dtype=torch.uint8).pin_memory()
 
     t = timeit.Timer(functools.partial(file_read, input_file, aio_handle, bounce_buffer))

diff --git a/deepnvme/file_access/aio_store_cpu_tensor.py b/deepnvme/file_access/aio_store_cpu_tensor.py
@@ -15,7 +15,7 @@ def main():
     file_sz = args.mb_size*(1024**2)
     app_tensor = torch.empty(file_sz, dtype=torch.uint8, device='cpu', requires_grad=False)
 
-    aio_handle = AsyncIOBuilder().load().aio_handle(1024**2, 128, True, True, 1)
+    aio_handle = AsyncIOBuilder().load().aio_handle()
     bounce_buffer = torch.empty(file_sz, dtype=torch.uint8, requires_grad=False).pin_memory()
 
 

diff --git a/deepnvme/file_access/aio_store_gpu_tensor.py b/deepnvme/file_access/aio_store_gpu_tensor.py
@@ -15,7 +15,7 @@ def main():
     file_sz = args.mb_size*(1024**2)
     app_tensor = torch.empty(file_sz, dtype=torch.uint8, device='cuda', requires_grad=False)
 
-    aio_handle = AsyncIOBuilder().load().aio_handle(1024**2, 128, True, True, 1)
+    aio_handle = AsyncIOBuilder().load().aio_handle()
     bounce_buffer = torch.empty(file_sz, dtype=torch.uint8, requires_grad=False).pin_memory()
 
 

diff --git a/deepnvme/file_access/gds_load_gpu_tensor.py b/deepnvme/file_access/gds_load_gpu_tensor.py
@@ -13,7 +13,7 @@ def main():
     file_sz = os.path.getsize(input_file)
     cnt = args.loop
 
-    gds_handle = GDSBuilder().load().gds_handle(1024**2, 128, True, True, 1)
+    gds_handle = GDSBuilder().load().gds_handle()
     gds_buffer = gds_handle.new_pinned_device_tensor(file_sz, torch.empty(0, dtype=torch.uint8, device='cuda', requires_grad=False))
 
     t = timeit.Timer(functools.partial(file_read, input_file, gds_handle, gds_buffer))

diff --git a/deepnvme/file_access/gds_store_gpu_tensor.py b/deepnvme/file_access/gds_store_gpu_tensor.py
@@ -15,7 +15,7 @@ def main():
     file_sz = args.mb_size*(1024**2)
     app_tensor = torch.empty(file_sz, dtype=torch.uint8, device='cuda', requires_grad=False)
 
-    gds_handle = GDSBuilder().load().gds_handle(1024**2, 128, True, True, 1)
+    gds_handle = GDSBuilder().load().gds_handle()
     gds_buffer = gds_handle.new_pinned_device_tensor(file_sz, torch.empty(0, dtype=torch.uint8, device='cuda', requires_grad=False))
 
     t = timeit.Timer(functools.partial(file_write, output_file, app_tensor, gds_handle, gds_buffer))