diff options
author | Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> | 2019-10-02 16:05:44 +0100 |
---|---|---|
committer | Kevin May <kevin.may@arm.com> | 2019-10-03 06:49:39 +0000 |
commit | 83450fa5d56cafe06d64fcd5f73dd9ed79c95e60 (patch) | |
tree | 431d8acfbc217b644d694d802ac67fe3f9032e78 | |
parent | f97f6da835802187da03f597dcc30386c7b9b15b (diff) | |
download | armnn-83450fa5d56cafe06d64fcd5f73dd9ed79c95e60.tar.gz |
IVGCVSW-3941 Fix bug around file opening in RecordByRecordCaffeParser
* Added error check after attempting to create input file stream and
throw FileNotFoundException in case of failure
Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Change-Id: I46b77e21c5330bb60e1190ddbefa6473e43085d9
-rw-r--r-- | src/armnnCaffeParser/RecordByRecordCaffeParser.cpp | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/src/armnnCaffeParser/RecordByRecordCaffeParser.cpp b/src/armnnCaffeParser/RecordByRecordCaffeParser.cpp index 592dc5c0de..cb7943655d 100644 --- a/src/armnnCaffeParser/RecordByRecordCaffeParser.cpp +++ b/src/armnnCaffeParser/RecordByRecordCaffeParser.cpp @@ -474,8 +474,12 @@ armnn::INetworkPtr RecordByRecordCaffeParser::CreateNetworkFromBinaryFile( } m_RequestedOutputs = requestedOutputs; - //FILE * fp = fopen(graphFile, "rb"); std::ifstream ifs(graphFile, std::ifstream::in|std::ifstream::binary); + if (ifs.fail()) + { + throw armnn::FileNotFoundException("Failed to open graph file '" + std::string(graphFile) + "'"); + } + std::vector<LayerParameterInfo> layerInfo; NetParameterInfo netParameterInfo; while(true) @@ -727,6 +731,3 @@ armnn::INetworkPtr RecordByRecordCaffeParser::LoadLayers(std::ifstream& ifs, return move(m_Network); } - - - |