Skip to content

Commit fc77109

Browse files
authored
Prevent stack-use-after-scope (NVIDIA#5572)
Signed-off-by: Joaquin Anton <janton@nvidia.com>
1 parent cc6a9b8 commit fc77109

File tree

1 file changed

+4
-5
lines changed

1 file changed

+4
-5
lines changed

dali/operators/io/file/file_read.cc

+4-5
Original file line numberDiff line numberDiff line change
@@ -39,16 +39,15 @@ class FileRead : public StatelessOperator<CPUBackend> {
3939
filenames_.resize(nsamples);
4040
files_.resize(nsamples);
4141
auto &tp = ws.GetThreadPool();
42+
FileStream::Options opts;
43+
opts.use_mmap = !dont_use_mmap_;
44+
opts.use_odirect = use_o_direct_;
45+
opts.read_ahead = false;
4246
for (int i = 0; i < nsamples; i++) {
4347
size_t filename_len = filepaths.tensor_shape_span(i)[0];
4448
filenames_[i].resize(filename_len + 1);
4549
std::memcpy(&filenames_[i][0], filepaths.raw_tensor(i), filename_len);
4650
filenames_[i][filename_len] = '\0';
47-
48-
FileStream::Options opts;
49-
opts.use_mmap = !dont_use_mmap_;
50-
opts.use_odirect = use_o_direct_;
51-
opts.read_ahead = false;
5251
tp.AddWork(
5352
[&, i](int tid) {
5453
files_[i] = FileStream::Open(filenames_[i], opts);

0 commit comments

Comments
 (0)