experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[XLA] Fix broken HLO test

Browse Source 
PiperOrigin-RevId: 352813527
Change-Id: Ia605e5ce9f033b397df202577fc403c5ba022d3a
This commit is contained in:
David Majnemer 2021-01-20 09:52:25 -08:00 committed by TensorFlower Gardener
parent a801850ccc
commit b64b75c35b
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/compiler/xla/tests/collective_ops_test.cc
View File

@ -770,7 +770,7 @@ XLA_TEST_F(CollectiveOpsTest, AllGather_Dim1) {
ENTRY test_computation { ENTRY test_computation {
id = u32[] replica-id() id = u32[] replica-id()
id2 = u32[2, 1] broadcast(id), dimensions={} id2 = u32[2, 1] broadcast(id), dimensions={}
a0 = u32[2, 1] constant({{10, 15}}) a0 = u32[2, 1] constant({10}, {15})
a1 = u32[2, 1] add(id2, a0) a1 = u32[2, 1] add(id2, a0)
allgather = u32[2, 4] all-gather(a1), dimensions={1} allgather = u32[2, 4] all-gather(a1), dimensions={1}
ROOT out = u32[8] reshape(allgather) ROOT out = u32[8] reshape(allgather)