zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-12-16 01:49:05 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Read arrays from files faster (#1330)

Browse Source 
* read faster

* faster write as well

* set default permission for linux

* comment
This commit is contained in:
Awni Hannun
2024-08-14 20:09:56 -07:00
committed by GitHub
parent 99bb7d3a58
commit d0630ffe8c
4 changed files with 57 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
mlx/backend/common/load.cpp
View File

@@ -33,7 +33,7 @@ void Load::eval(const std::vector<array>& inputs, array& out) {
assert(inputs.size() == 0);
out.set_data(allocator::malloc_or_wait(out.nbytes()));
reader_->seek(offset_, std::ios_base::beg);
reader_->seek(offset_);
reader_->read(out.data<char>(), out.nbytes());
if (swap_endianness_) {