refine device_context_test

8 years ago · 2403045cbd
parent f168843e47
commit 2403045cbd
1 changed files with 1 additions and 0 deletions
--- a/paddle/platform/device_context_test.cc
+++ b/paddle/platform/device_context_test.cc
@ -45,6 +45,7 @@ TEST(Device, CUDADeviceContext) {
    ASSERT_NE(nullptr, cublas_handle);
    curandGenerator_t curand_handle = device_context->curand_generator();
    ASSERT_NE(nullptr, curand_handle);
+    ASSERT_NE(nullptr, device_context->stream());
    delete device_context;
  }
 }