提交 64095260 编写于 作者: R Ray Liu 提交者: GitHub

Merge pull request #1405 from xiebaiyuan/develop

support gpu mobilenet single build fixed #1404
...@@ -13,7 +13,7 @@ See the License for the specific language governing permissions and ...@@ -13,7 +13,7 @@ See the License for the specific language governing permissions and
limitations under the License. */ limitations under the License. */
#ifdef RELU_OP #ifdef RELU_OP
#include "operators/kernel/relu_kernel.h" #include "operators/kernel/activation_kernel.h"
namespace paddle_mobile { namespace paddle_mobile {
namespace operators { namespace operators {
......
...@@ -30,6 +30,10 @@ if (CON GREATER -1) ...@@ -30,6 +30,10 @@ if (CON GREATER -1)
target_link_libraries(test-mobilenet-combine paddle-mobile) target_link_libraries(test-mobilenet-combine paddle-mobile)
set(FOUND_MATCH ON) set(FOUND_MATCH ON)
# gen test
ADD_EXECUTABLE(test-mobilenetgpu net/test_mobilenet_GPU.cpp test_helper.h test_include.h)
target_link_libraries(test-mobilenetgpu paddle-mobile)
endif () endif ()
list(FIND NET "yolo" CON) list(FIND NET "yolo" CON)
......
...@@ -25,11 +25,11 @@ int main() { ...@@ -25,11 +25,11 @@ int main() {
paddle_mobile.SetCLPath("/data/local/tmp/bin"); paddle_mobile.SetCLPath("/data/local/tmp/bin");
#endif #endif
auto isok = // auto isok =
paddle_mobile.Load(std::string(g_mobilenet_mul) + "/model", // paddle_mobile.Load(std::string(g_mobilenet_mul) + "/model",
std::string(g_mobilenet_mul) + "/params", true); // 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), true);
if (isok) { if (isok) {
auto time2 = paddle_mobile::time(); auto time2 = paddle_mobile::time();
std::cout << "load cost :" << paddle_mobile::time_diff(time1, time2) << "ms" std::cout << "load cost :" << paddle_mobile::time_diff(time1, time2) << "ms"
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册