Update test_op.h

Change 0 to opContext_.run_ctx.ctx.dev_id
apache · Mar 1, 2019 · 7bdd4d4 · 7bdd4d4
1 parent d4ecd4a
commit 7bdd4d4
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, 0);
+    opContext_.run_ctx.stream = mshadow::NewStream<gpu>(true, true, opContext_.run_ctx.ctx.dev_id);
     CHECK_EQ(opContext_.run_ctx.stream != nullptr, true)
       << "Unable to allocate a GPU stream";
   }