提交 b95834dc 编写于 作者: T tensor-tang

disable use_gpu when test mkl recurrent layer comparing with cpu

上级 0f8aad29
...@@ -551,6 +551,8 @@ TEST(MKLPackedLayer, RecurrentLayer) { ...@@ -551,6 +551,8 @@ TEST(MKLPackedLayer, RecurrentLayer) {
layerConfig2.set_type("mkl_packed_recurrent"); layerConfig2.set_type("mkl_packed_recurrent");
layerConfig2.set_active_type("relu"); layerConfig2.set_active_type("relu");
FLAGS_use_gpu = false;
for (auto layerSize : {32, 64, 128, 256, 512}) { for (auto layerSize : {32, 64, 128, 256, 512}) {
for (auto batchSize : {1, 5, 100, 500}) { for (auto batchSize : {1, 5, 100, 500}) {
for (auto reversed : {true, false}) { for (auto reversed : {true, false}) {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册