diff --git a/deepnvme/file_access/aio_load_cpu_tensor.py b/deepnvme/file_access/aio_load_cpu_tensor.py index c75d1ba7c..27a1e61c5 100644 --- 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 index 5720676a6..aeecc6e5d 100644 --- 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 index 6f257e5a2..20c03792b 100644 --- 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 index d00ee17c3..71a4aa7bb 100644 --- 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 index a9220517f..dd6273707 100644 --- 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 index 0b1b9a542..06ba508ba 100644 --- 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))