From 2403045cbd57eb837d5ab82e2acc66767c1d3224 Mon Sep 17 00:00:00 2001
From: qijun <qijun1994@hotmail.com>
Date: Tue, 15 Aug 2017 07:03:54 +0000
Subject: [PATCH] refine device_context_test

---
 paddle/platform/device_context_test.cc | 1 +
 1 file changed, 1 insertion(+)

diff --git a/paddle/platform/device_context_test.cc b/paddle/platform/device_context_test.cc
index 65345c433c0..8b764bdcd9d 100644
--- 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;
   }
 }
-- 
GitLab