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

Method prefetch

mlx/data/Stream.cpp:115–118  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

113}
114
115Stream Stream::prefetch(int prefetch_size, int num_thread) const {
116 return Stream(
117 std::make_shared<stream::Prefetch>(self_, prefetch_size, num_thread));
118}
119Stream Stream::prefetch_if(bool cond, int prefetch_size, int num_thread) const {
120 if (cond) {
121 return prefetch(prefetch_size, num_thread);

Callers 7

load_librispeechFunction · 0.45
iterateFunction · 0.45
iterateFunction · 0.45
iterateFunction · 0.45
iterateFunction · 0.45
iterateFunction · 0.45
iterateFunction · 0.45

Calls 1

StreamClass · 0.70

Tested by

no test coverage detected