From d00d01e1a8082023ea776e1e80cbd1fa4834dcc7 Mon Sep 17 00:00:00 2001 From: Lai Wei Date: Wed, 27 Feb 2019 22:15:39 -0800 Subject: [PATCH 1/5] use cudnn for dropout by default --- src/operator/nn/dropout-inl.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/operator/nn/dropout-inl.h b/src/operator/nn/dropout-inl.h index 2a828994fb44..2b88eec20820 100644 --- a/src/operator/nn/dropout-inl.h +++ b/src/operator/nn/dropout-inl.h @@ -77,7 +77,7 @@ struct DropoutParam : public dmlc::Parameter { .describe("Whether to only turn on dropout during training or to also turn on for inference."); DMLC_DECLARE_FIELD(axes).set_default(TShape()) .describe("Axes for variational dropout kernel."); - DMLC_DECLARE_FIELD(cudnn_off).set_default(dmlc::optional(true)) + DMLC_DECLARE_FIELD(cudnn_off).set_default(dmlc::optional(false)) .describe("Whether to turn off cudnn in dropout operator. " "This option is ignored if axes is specified."); } From d4ecd4a3b8f2da484cbe7fe5e1fb0b6740ccaa68 Mon Sep 17 00:00:00 2001 From: Lai Wei Date: Thu, 28 Feb 2019 10:46:37 -0800 Subject: [PATCH 2/5] update test --- tests/cpp/include/test_op.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/cpp/include/test_op.h b/tests/cpp/include/test_op.h index 7a0c6d3878ee..cb45c89bc684 100644 --- a/tests/cpp/include/test_op.h +++ b/tests/cpp/include/test_op.h @@ -72,7 +72,7 @@ struct GPUStreamScope { : opContext_(*opContext) { CHECK_EQ(opContext_.run_ctx.stream == nullptr, true) << "Invalid runtime context stream state"; - opContext_.run_ctx.stream = mshadow::NewStream(true, true); + opContext_.run_ctx.stream = mshadow::NewStream(true, true, 0); CHECK_EQ(opContext_.run_ctx.stream != nullptr, true) << "Unable to allocate a GPU stream"; } From 0a5ea7c0f4af57031460cefa91f0903337ae1f94 Mon Sep 17 00:00:00 2001 From: Lai Wei Date: Fri, 1 Mar 2019 14:31:32 -0800 Subject: [PATCH 3/5] use dev_id --- tests/cpp/include/test_op.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/cpp/include/test_op.h b/tests/cpp/include/test_op.h index cb45c89bc684..17f2e568a54d 100644 --- a/tests/cpp/include/test_op.h +++ b/tests/cpp/include/test_op.h @@ -72,7 +72,7 @@ struct GPUStreamScope { : opContext_(*opContext) { CHECK_EQ(opContext_.run_ctx.stream == nullptr, true) << "Invalid runtime context stream state"; - opContext_.run_ctx.stream = mshadow::NewStream(true, true, 0); + opContext_.run_ctx.stream = mshadow::NewStream(true, true, opContext_.run_ctx.ctx.dev_id); CHECK_EQ(opContext_.run_ctx.stream != nullptr, true) << "Unable to allocate a GPU stream"; } From 725cef6e784f0f83cc7362c6d97401e3b098b546 Mon Sep 17 00:00:00 2001 From: Lai Wei Date: Sat, 2 Mar 2019 20:39:58 -0800 Subject: [PATCH 4/5] trigger ci From 2a719daa5206485c413cc8b6574affcaa6f08cea Mon Sep 17 00:00:00 2001 From: Lai Wei Date: Mon, 4 Mar 2019 18:41:23 -0800 Subject: [PATCH 5/5] trigger ci