diff options
| author |  Yunxing Dai <yunxing@google.com> | 2018-08-23 14:53:20 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-23 14:57:41 -0700 |
| commit | a73bc982669c8eb4bec9418a94a96b64551e641b (patch) | |
| tree | 663477d414aad7a982acecbec029a814cd74d9b4 /tensorflow/compiler/xla/service/hlo_sharding_test.cc | |
| parent | c6b8f3617662ff0223eace9774089ce509867160 (diff) | |
Do not crash when an empty tuple is passed into hlo_sharding.
PiperOrigin-RevId: 210005372
Diffstat (limited to 'tensorflow/compiler/xla/service/hlo_sharding_test.cc')
| -rw-r--r-- | tensorflow/compiler/xla/service/hlo_sharding_test.cc | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/service/hlo_sharding_test.cc b/tensorflow/compiler/xla/service/hlo_sharding_test.cc index 45fc300fca..2341f8ada0 100644 --- a/tensorflow/compiler/xla/service/hlo_sharding_test.cc +++ b/tensorflow/compiler/xla/service/hlo_sharding_test.cc @@ -115,6 +115,13 @@ TEST_F(HloShardingTest, Tile) { } } +// Tests that empty tuple is supported. +TEST_F(HloShardingTest, EmptySingleTuple) { + HloSharding sharding = HloSharding::SingleTuple(ShapeUtil::MakeTupleShape({}), + HloSharding::AssignDevice(0)); + EXPECT_TRUE(sharding.ExtractSingleSharding()); +} + TEST_F(HloShardingTest, NestedTuple) { // nested_tuple_shape = (f32[], (f32[3]), f32[4, 6]) Shape nested_tuple_shape = ShapeUtil::MakeTupleShape({ |