From b857488021bc00af9f7460eba6f396047b159c7f Mon Sep 17 00:00:00 2001 From: drebain Date: Mon, 24 Aug 2020 12:12:55 -0700 Subject: [PATCH] Make testMultiGPU pass when on CPU only --- tensorflow/python/kernel_tests/dynamic_partition_op_test.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tensorflow/python/kernel_tests/dynamic_partition_op_test.py b/tensorflow/python/kernel_tests/dynamic_partition_op_test.py index 2ff0a1d631e..df639d3d290 100644 --- a/tensorflow/python/kernel_tests/dynamic_partition_op_test.py +++ b/tensorflow/python/kernel_tests/dynamic_partition_op_test.py @@ -358,7 +358,8 @@ class DynamicPartitionTest(test.TestCase): partitions = constant_op.constant(np.arange(1000, dtype=np.int32) % 10) result = data_flow_ops.dynamic_partition(data, partitions, 10) results.append(self.evaluate(result)) - self.assertAllEqual(results, np.zeros((len(device_list), 10, 100))) + if device_list: + self.assertAllEqual(results, np.zeros((len(device_list), 10, 100))) if __name__ == "__main__":