saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-29 20:14:01 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Close input stream too

Browse source
This commit is contained in:
Ryan Lai 2020-07-20 19:32:34 -07:00
parent 846e550c4f
commit 4564776733
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
winml/test/common/protobufHelpers.cpp
View file

@ -65,7 +65,7 @@ bool ProtobufHelpers::LoadOnnxTensorFromProtobufFile(onnx::TensorProto& tensor,
google::protobuf::io::FileInputStream input(fd, block_size);
const bool result = tensor.ParseFromZeroCopyStream(&input) && input.GetErrno() == 0;
status = onnxruntime::Env::Default().FileClose(fd);
if (!result || !status.IsOK()) {
if (!result || !status.IsOK() || !input.Close()) {
return false;
}
return true;