OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

Expose recordSize in ChunkRecordIterator (#120239)

Browse Source 
Summary: Add a public method to read recordSize in ChunkRecordIterator

Test Plan: ci

Differential Revision: D53931944

Pull Request resolved: https://github.com/pytorch/pytorch/pull/120239
Approved by: https://github.com/zoranzhao
This commit is contained in:
Quinn Zhu 2024-02-21 04:33:03 +00:00 committed by PyTorch MergeBot
parent 26610175d2
commit 3993771617
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
caffe2/serialize/inline_container.h
View File

@ -104,6 +104,7 @@ class TORCH_API ChunkRecordIterator {
// Read at most `chunkSize` into `buf`. Return the number of actual bytes read.
size_t next(void* buf);
size_t recordSize() const { return recordSize_; }
private:
ChunkRecordIterator(