diff --git a/src/io.cpp b/src/io.cpp index f0c625b9150bc202297e14fcb81ac4c9fe6e5c7d..0c06f76ca33133fee9a3d3b618d8f26714631a72 100644 --- a/src/io.cpp +++ b/src/io.cpp @@ -12,20 +12,24 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions and limitations under the License. */ +#include "io.h" +#include #include #include "common/log.h" -#include "framework/framework.pb.h" -#include "framework/lod_tensor.h" -#include "framework/program/program_desc.h" +#include "common/enforce.h" #include "framework/scope.h" #include "framework/tensor.h" -#include "io.h" +#include "framework/lod_tensor.h" +#include "framework/framework.pb.h" +#include "framework/program/program_desc.h" namespace paddle_mobile { void ReadBinaryFile(const std::string &filename, std::string *contents) { std::ifstream fin(filename, std::ios::in | std::ios::binary); + PADDLE_MOBILE_ENFORCE(fin.is_open(), + "open file: %s failed", filename.c_str()); fin.seekg(0, std::ios::end); contents->clear(); contents->resize(fin.tellg()); @@ -38,7 +42,8 @@ template void Loader::LoadVar(framework::LoDTensor *tensor, const std::string &file_path) { std::ifstream is(file_path); - + PADDLE_MOBILE_ENFORCE(is.is_open(), + "open file: %s failed", file_path.c_str()); std::fpos pos; pos = is.tellg(); // save current position is.seekg(0, std::ios::end); @@ -238,6 +243,8 @@ const framework::Program Loader::Load( var.type().type() != framework::proto::VarType::FETCH_LIST) { std::string file_path = dirname + "/" + var.name(); std::ifstream is(file_path); + PADDLE_MOBILE_ENFORCE(is.is_open(), + "open file: %s failed", file_path.c_str()); std::fpos pos; pos = is.tellg(); // save current position is.seekg(0, std::ios::end); diff --git a/src/io.h b/src/io.h index 7a8d5863ebae90861d0b6d999a185e5755b6b57f..bbc0c0c6ebdf29012240e4f9e1fc62a16c174a4e 100644 --- a/src/io.h +++ b/src/io.h @@ -18,8 +18,8 @@ limitations under the License. */ #include "common/types.h" #include "framework/lod_tensor.h" -#include "framework/paddle_mobile_object.h" #include "framework/program/program.h" +#include "framework/paddle_mobile_object.h" namespace paddle_mobile {