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."); } diff --git a/tests/cpp/include/test_op.h b/tests/cpp/include/test_op.h index 7a0c6d3878ee..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); + 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"; }