Implement Tensor Memory allocator #89

edgarriba · 2024-07-02T15:19:49Z

implement TensorStorage and TensorAllocator
the TensorStorage is implemented in terms of arrow::buffer::Buffer https://docs.rs/arrow/latest/arrow/buffer/struct.Buffer.html

src/tensor/allocator.rs

src/tensor/base.rs

src/tensor/serde.rs

src/tensor/storage.rs

edgarriba added 6 commits July 2, 2024 11:40

implement storage and allocator

46c01dd

small fixes

7f78053

add serialization

3331b72

fixes

3e10676

fix tests

7fcd341

more errors

b951175

github-actions bot reviewed Jul 2, 2024

View reviewed changes

edgarriba added 2 commits July 3, 2024 13:25

implement tensor with arrow Buffer

40696a3

remove into_raw_vec

10b62b9

edgarriba changed the title ~~[wip] Implement Tensor Memory allocator~~ Implement Tensor Memory allocator Jul 3, 2024

edgarriba marked this pull request as ready for review July 3, 2024 15:56

docs

14f12bf

edgarriba merged commit 5fef6d5 into main Jul 3, 2024
9 checks passed

edgarriba deleted the memory-allocator branch January 2, 2025 06:56

Provide feedback