MCPcopy Create free account
hub / github.com/ml-explore/mlx-data / buffered

Method buffered

mlx/data/Stream.cpp:41–51  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

39}
40
41Stream Stream::buffered(
42 int64_t buffer_size,
43 std::function<Buffer(const Buffer)> on_refill,
44 int num_thread) const {
45 auto on_refill_stream =
46 [on_refill](const std::shared_ptr<buffer::Buffer>& buf) {
47 return on_refill(Buffer(buf)).self_;
48 };
49 return Stream(std::make_shared<stream::CallbackBuffered>(
50 self_, buffer_size, on_refill_stream, num_thread));
51};
52
53Stream Stream::csv_reader_from_key(
54 const std::string& key,

Callers 1

init_mlx_data_streamFunction · 0.80

Calls 2

BufferClass · 0.70
StreamClass · 0.70

Tested by

no test coverage detected