File tree 2 files changed +4
-3
lines changed
2 files changed +4
-3
lines changed Original file line number Diff line number Diff line change 1
1
#pragma once
2
2
#include < caffe2/serialize/read_adapter_interface.h>
3
+ #include < sys/types.h>
3
4
#include < cstdint>
4
5
#include < cstring>
5
- #include < sys/types.h>
6
6
7
7
namespace caffe2 ::serialize {
8
8
Original file line number Diff line number Diff line change @@ -400,7 +400,8 @@ size_t PyTorchStreamReader::getRecordMultiReaders(
400
400
}
401
401
readSizes[i] = size;
402
402
LOG (INFO) << " Thread " << i << " read [" << startPos << " -" << endPos
403
- << " ] " << " from " << name << " of size " << n;
403
+ << " ] "
404
+ << " from " << name << " of size " << n;
404
405
TORCH_CHECK (
405
406
threadReadSize == size,
406
407
" record size " ,
@@ -697,7 +698,7 @@ void PyTorchStreamWriter::setup(const string& file_name) {
697
698
698
699
const std::string dir_name = parentdir (file_name);
699
700
if (!dir_name.empty ()) {
700
- struct stat st{};
701
+ struct stat st {};
701
702
bool dir_exists =
702
703
(stat (dir_name.c_str (), &st) == 0 && (st.st_mode & S_IFDIR));
703
704
TORCH_CHECK (
You can’t perform that action at this time.
0 commit comments