From fa716885c8621dddd938fab66ebf1b93ebf55b1a Mon Sep 17 00:00:00 2001 From: yangfei Date: Tue, 6 Nov 2018 13:30:01 +0800 Subject: [PATCH] repair bug of test-gpu --- test/net/test_mobilenet_GPU.cpp | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/test/net/test_mobilenet_GPU.cpp b/test/net/test_mobilenet_GPU.cpp index a3e61f5edd..e410baf776 100644 --- a/test/net/test_mobilenet_GPU.cpp +++ b/test/net/test_mobilenet_GPU.cpp @@ -13,22 +13,23 @@ See the License for the specific language governing permissions and limitations under the License. */ #include +#include "../../src/common/types.h" #include "../test_helper.h" #include "../test_include.h" -#include "../../src/common/types.h" int main() { paddle_mobile::PaddleMobile paddle_mobile; // paddle_mobile.SetThreadNum(4); auto time1 = paddle_mobile::time(); #ifdef PADDLE_MOBILE_CL - paddle_mobile.SetCLPath("/data/local/tmp/bin"); + paddle_mobile.SetCLPath("/data/local/tmp/bin"); #endif - auto isok = paddle_mobile.Load(std::string(g_mobilenet_mul) + "/model", - std::string(g_mobilenet_mul) + "/params", true); + auto isok = + paddle_mobile.Load(std::string(g_mobilenet_mul) + "/model", + std::string(g_mobilenet_mul) + "/params", true); -// auto isok = paddle_mobile.Load(std::string(g_mobilenet_mul), true); + // auto isok = paddle_mobile.Load(std::string(g_mobilenet_mul), true); if (isok) { auto time2 = paddle_mobile::time(); std::cout << "load cost :" << paddle_mobile::time_diff(time1, time2) << "ms" -- GitLab