diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-01-16 05:15:00 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-01-16 05:18:21 -0800 |
commit | 6e259f0a716c124b16c9a2aaddd3ee6c9edc5db9 (patch) | |
tree | 2d32b97129bbd4ae6c6a2f933424c6dba03f6be1 /tensorflow/contrib/nccl | |
parent | 95327da03d2e277f71a11acece33f402c6bb3b85 (diff) |
Fixes nccl_ops_test error introduced in CL 181736230.
PiperOrigin-RevId: 182039394
Diffstat (limited to 'tensorflow/contrib/nccl')
-rw-r--r-- | tensorflow/contrib/nccl/python/ops/nccl_ops_test.py | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/tensorflow/contrib/nccl/python/ops/nccl_ops_test.py b/tensorflow/contrib/nccl/python/ops/nccl_ops_test.py index 436ebf1da7..98fe394c5b 100644 --- a/tensorflow/contrib/nccl/python/ops/nccl_ops_test.py +++ b/tensorflow/contrib/nccl/python/ops/nccl_ops_test.py @@ -172,8 +172,7 @@ class BroadcastTest(NcclTestCase): (['/device:GPU:0', '/device:CPU:0'],)) except errors.NotFoundError as e: self.assertRegexpMatches( - e.value, - "No registered '_NcclBroadcastRecv' OpKernel for CPU devices") + str(e), "No registered '_NcclBroadcastRecv' OpKernel for CPU devices") else: # Session isn't executed when no GPU is available. if test.is_gpu_available(): |