From b95834dc0c43cedd27124e18a13345712dcf9d47 Mon Sep 17 00:00:00 2001 From: tensor-tang Date: Wed, 20 Dec 2017 09:05:41 +0800 Subject: [PATCH] disable use_gpu when test mkl recurrent layer comparing with cpu --- paddle/gserver/tests/test_RecurrentLayer.cpp | 2 ++ 1 file changed, 2 insertions(+) diff --git a/paddle/gserver/tests/test_RecurrentLayer.cpp b/paddle/gserver/tests/test_RecurrentLayer.cpp index 1f3115857..44d84dd8b 100644 --- a/paddle/gserver/tests/test_RecurrentLayer.cpp +++ b/paddle/gserver/tests/test_RecurrentLayer.cpp @@ -551,6 +551,8 @@ TEST(MKLPackedLayer, RecurrentLayer) { layerConfig2.set_type("mkl_packed_recurrent"); layerConfig2.set_active_type("relu"); + FLAGS_use_gpu = false; + for (auto layerSize : {32, 64, 128, 256, 512}) { for (auto batchSize : {1, 5, 100, 500}) { for (auto reversed : {true, false}) { -- GitLab