From 6a467389863aa360cf09715dc5c0301805abb44e Mon Sep 17 00:00:00 2001 From: Zhizhen Qin Date: Thu, 14 Feb 2019 17:22:34 -0800 Subject: [PATCH] Fix android crash when model detects nothing Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/17119 Reviewed By: sf-wind Differential Revision: D14087835 Pulled By: ZhizhenQin fbshipit-source-id: 32e61d46679bae645fd0bbec724513cfa5c553ab --- binaries/benchmark_helper.h | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/binaries/benchmark_helper.h b/binaries/benchmark_helper.h index 8c24a3d..d09ba4e 100644 --- a/binaries/benchmark_helper.h +++ b/binaries/benchmark_helper.h @@ -82,7 +82,9 @@ void writeTextOutput( } // remove the last , string str = line.str(); - str.pop_back(); + if(str.length() != 0) { + str.pop_back(); + } lines.push_back(str); // static casts are workaround for MSVC build -- 2.7.4