update test

apache · Feb 28, 2019 · d4ecd4a · d4ecd4a
1 parent d00d01e
commit d4ecd4a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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<gpu>(true, true);
+    opContext_.run_ctx.stream = mshadow::NewStream<gpu>(true, true, 0);
     CHECK_EQ(opContext_.run_ctx.stream != nullptr, true)
       << "Unable to allocate a GPU stream";
   }