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__":