diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-06-14 18:27:53 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-06-14 18:30:35 -0700 |
| commit | 271c1a15f206ccae3762a76b0e47d2ae477d4863 (patch) | |
| tree | 0833f54f07e96949a7ed8f6fa47f6eb09d3996d1 /tensorflow/compiler/xla/service/hlo_parser_test.cc | |
| parent | 332c4d699c23b8d6f8b17b48600f831cacad4aae (diff) | |
Split out HloAllReduceInstruction as a subclass of HloInstruction.
HloAllReduceInstruction can't subclass HloSendRecvInstruction because channel_id was optional in all reduce. So add 'all_reduce_id' instead.
PiperOrigin-RevId: 200653920
Diffstat (limited to 'tensorflow/compiler/xla/service/hlo_parser_test.cc')
| -rw-r--r-- | tensorflow/compiler/xla/service/hlo_parser_test.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/compiler/xla/service/hlo_parser_test.cc b/tensorflow/compiler/xla/service/hlo_parser_test.cc index f834d34d57..d551400d1e 100644 --- a/tensorflow/compiler/xla/service/hlo_parser_test.cc +++ b/tensorflow/compiler/xla/service/hlo_parser_test.cc @@ -913,7 +913,7 @@ add { ENTRY CRS { input = f32[8]{0} parameter(0) - ROOT crs = f32[8]{0} cross-replica-sum(input), to_apply=add + ROOT crs = f32[8]{0} cross-replica-sum(input), replica_group_ids={}, to_apply=add } )" @@ -931,7 +931,7 @@ add { ENTRY CrossReplicaSumWithSubgroups { input = f32[128,32]{0,1} parameter(0) - ROOT cross-replica-sum = f32[128,32]{0,1} cross-replica-sum(input), to_apply=add, replica_group_ids={0,0,1,1}, barrier="abc" + ROOT cross-replica-sum = f32[128,32]{0,1} cross-replica-sum(input), replica_group_ids={0,0,1,1}, barrier="abc", to_apply=add } )" |