From 099d1b87f62e9b2643bb5e4c57ba9322af867598 Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Tue, 28 Dec 2021 18:09:16 +0800
Subject: [PATCH] fix(lite/load_and_run): fix suffix parse error

GitOrigin-RevId: 8a9dac219e62ca53c33a6f097eba57b62051c8e2
---
 lite/load_and_run/src/helpers/data_parser.cpp | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

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")) {