From 557f3895b093569c3677fd7cfc50e72c9496c94a Mon Sep 17 00:00:00 2001 From: botbw Date: Mon, 14 Oct 2024 07:12:03 +0000 Subject: [PATCH] [fio] use raw tensor ptr instead of numpy --- tensornvme/async_file_io.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tensornvme/async_file_io.py b/tensornvme/async_file_io.py index cdf52dd..5ee073b 100644 --- a/tensornvme/async_file_io.py +++ b/tensornvme/async_file_io.py @@ -40,8 +40,9 @@ def save( self.write(header_bytes) for tensor in tensors: - tensor_bytes = _tobytes(tensor, '') - self.write(tensor_bytes) + self.io.write(tensor.data_ptr(), tensor.numel() * tensor.element_size(), self.offset) + self.offset += tensor.numel() * tensor.element_size() + self.buffers.append(tensor) def flush(self) -> None: pass