aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/compiler/xla/service/hlo_parser_test.cc
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2018-06-14 18:27:53 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-06-14 18:30:35 -0700
commit271c1a15f206ccae3762a76b0e47d2ae477d4863 (patch)
tree0833f54f07e96949a7ed8f6fa47f6eb09d3996d1 /tensorflow/compiler/xla/service/hlo_parser_test.cc
parent332c4d699c23b8d6f8b17b48600f831cacad4aae (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.cc4
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
}
)"