MCPcopy Create free account
hub / github.com/NVIDIA/DALI / ReadHeader

Method ReadHeader

dali/operators/reader/loader/numpy_loader_gpu.cc:37–54  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

35}
36
37void NumpyFileWrapperGPU::ReadHeader(detail::NumpyHeaderCache &cache) {
38 numpy::HeaderData header;
39 bool ret = cache.GetFromCache(filename, header);
40 try {
41 if (!ret) {
42 numpy::ParseHeader(header, file_stream_.get());
43 cache.UpdateCache(filename, header);
44 }
45 } catch (const std::runtime_error &e) {
46 DALI_FAIL(e.what(), ". File: ", filename);
47 }
48 file_stream_->SeekRead(header.data_offset);
49
50 type = header.type();
51 shape = header.shape;
52 fortran_order = header.fortran_order;
53 data_offset = header.data_offset;
54}
55
56void NumpyFileWrapperGPU::ReadRawChunk(void* buffer, size_t bytes,
57 Index buffer_offset, Index file_offset) {

Callers 1

PrefetchMethod · 0.80

Calls 7

GetFromCacheMethod · 0.80
UpdateCacheMethod · 0.80
ParseHeaderFunction · 0.50
getMethod · 0.45
whatMethod · 0.45
SeekReadMethod · 0.45
typeMethod · 0.45

Tested by

no test coverage detected