diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-08-28 11:37:18 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-28 11:41:45 -0700 |
commit | 6de10fb253098c9ff65e9d4083c4de84f3ff5f76 (patch) | |
tree | af6c7c66f13892c9df05e55c83896ae4bd67a77e /tensorflow/compiler/xla/service/hlo_parser_test.cc | |
parent | 13c7499d5454b870eb3604d6b0ca241685cabe18 (diff) |
[XLA] Add the xla interface for CollectivePermute.
PiperOrigin-RevId: 210576458
Diffstat (limited to 'tensorflow/compiler/xla/service/hlo_parser_test.cc')
-rw-r--r-- | tensorflow/compiler/xla/service/hlo_parser_test.cc | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/service/hlo_parser_test.cc b/tensorflow/compiler/xla/service/hlo_parser_test.cc index b3d3ccda74..b1ef288b8e 100644 --- a/tensorflow/compiler/xla/service/hlo_parser_test.cc +++ b/tensorflow/compiler/xla/service/hlo_parser_test.cc @@ -1098,6 +1098,18 @@ ENTRY AllToAllWithSubgroups { )" }, +// collective-permute +{ +"CollectivePermute", +R"(HloModule CollectivePermute + +ENTRY CollectivePermute { + input = f32[128,32]{0,1} parameter(0) + ROOT root = f32[128,32]{0,1} collective-permute(input), source_target_pairs={{0,1},{1,2},{2,3}} +} + +)" +}, // Iota { "Iota", |