experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Assert return value of InsertTfPlatformGpuIdPair in

Browse Source 
tensorflow/core/grappler/clusters:utils_test test.

PiperOrigin-RevId: 226399061
This commit is contained in:
Anna R 2018-12-20 15:01:11 -08:00 committed by TensorFlower Gardener
parent 5ff27167b2
commit fdb5e7c3ea
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tensorflow/core/grappler/clusters/utils_test.cc
View File

@ -17,6 +17,7 @@ limitations under the License.
#include "tensorflow/core/common_runtime/gpu/gpu_id.h"
#include "tensorflow/core/common_runtime/gpu/gpu_id_manager.h"
#include "tensorflow/core/lib/core/status_test_util.h"
#include "tensorflow/core/platform/logging.h"
#include "tensorflow/core/platform/test.h"
#include "tensorflow/core/protobuf/device_properties.pb.h"
@ -82,12 +83,14 @@ TEST(UtilsTest, GetDeviceInfo) {
#if GOOGLE_CUDA
// Invalid platform GPU id.
GpuIdManager::InsertTfPlatformGpuIdPair(TfGpuId(0), PlatformGpuId(100));
TF_ASSERT_OK(
GpuIdManager::InsertTfPlatformGpuIdPair(TfGpuId(0), PlatformGpuId(100)));
properties = GetDeviceInfo(device);
EXPECT_EQ("UNKNOWN", properties.type());
// Valid platform GPU id.
GpuIdManager::InsertTfPlatformGpuIdPair(TfGpuId(1), PlatformGpuId(0));
TF_ASSERT_OK(
GpuIdManager::InsertTfPlatformGpuIdPair(TfGpuId(1), PlatformGpuId(0)));
device.id = 1;
properties = GetDeviceInfo(device);
EXPECT_EQ("GPU", properties.type());