diff --git a/lite/load_and_run/src/helpers/data_parser.cpp b/lite/load_and_run/src/helpers/data_parser.cpp index 0ba71626..d5b7518e 100644 --- a/lite/load_and_run/src/helpers/data_parser.cpp +++ b/lite/load_and_run/src/helpers/data_parser.cpp @@ -30,7 +30,12 @@ void DataParser::feed(const std::string& path) { } auto endWith = [blob_string](std::string suffix) -> bool { - return blob_string.rfind(suffix) == (blob_string.length() - suffix.length()); + const auto index = blob_string.rfind(suffix); + if (index != std::string::npos and + index == blob_string.length() - suffix.length()) { + return true; + } + return false; }; if (endWith(".ppm") || endWith(".pgm")) {