diff options
author | HyoukJoong Lee <hyouklee@google.com> | 2018-08-22 09:45:10 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-22 09:49:44 -0700 |
commit | 75ef667e5ab2075ee7a451cb2e46d6fa12dd9e59 (patch) | |
tree | 597d7409a50d4361ba58c4808ca9ca07560b5170 /tensorflow/compiler/xla/service/hlo_parser_test.cc | |
parent | 567189980f7a1c2aa09a5170bd8d01a6ec37d303 (diff) |
Change subgroup interface for CrossReplicaSum
PiperOrigin-RevId: 209780185
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 5a1993a3bb..f52cfadb81 100644 --- a/tensorflow/compiler/xla/service/hlo_parser_test.cc +++ b/tensorflow/compiler/xla/service/hlo_parser_test.cc @@ -1052,7 +1052,7 @@ add { ENTRY CRS { input = f32[8]{0} parameter(0) - ROOT crs = f32[8]{0} cross-replica-sum(input), replica_group_ids={}, to_apply=add + ROOT crs = f32[8]{0} cross-replica-sum(input), replica_groups={}, to_apply=add } )" @@ -1070,7 +1070,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), replica_group_ids={0,0,1,1}, barrier="abc", to_apply=add + ROOT cross-replica-sum = f32[128,32]{0,1} cross-replica-sum(input), replica_groups={{0,1},{2,3}}, barrier="abc", to_apply=add } )" |