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

[XLA][MLIR] Update IotaAddMultiply GPU test.

Browse Source 
PiperOrigin-RevId: 307552390
Change-Id: I7f9ea293f8a04a1d8253ab399f1315f8470459da
This commit is contained in:
Alexander Belyaev 2020-04-21 00:05:22 -07:00 committed by TensorFlower Gardener
parent df6853ac5b
commit adbacb4206
2 changed files with 5 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/compiler/xla/service/mlir_gpu/tests/BUILD
View File

@ -25,8 +25,6 @@ glob_lit_tests(
],
driver = "@llvm-project//mlir:run_lit.sh",
exclude = [
# TODO(b/149302060) Reenable once fusion is fixed.
"iota_add_multiply.hlo",
# TODO(b/137624192): Reenable once we can fuse reductions.
"fused_reduce.hlo",
],

10
tensorflow/compiler/xla/service/mlir_gpu/tests/iota_add_multiply.hlo → tensorflow/compiler/xla/service/mlir_gpu/tests/iota_add_subtract.hlo
View File

@ -1,16 +1,16 @@
// RUN: xla-gpu-opt -lowering-stage=GPU %s | FileCheck %s -dump-input-on-failure
HloModule AddMultiply
HloModule AddSubtract
ENTRY %AddMultiply (x: s32[2,2], y: s32[2,2]) -> s32[2,2] {
ENTRY %AddSubtract (x: s32[2,2], y: s32[2,2]) -> s32[2,2] {
%x = s32[2,2]{1,0} parameter(0)
%y = s32[2,2]{1,0} parameter(1)
%add = s32[2,2]{1,0} add(s32[2,2]{1,0} %x, s32[2,2]{1,0} %y)
%iota = s32[2, 2]{1,0} iota(), iota_dimension=0
ROOT %mul = s32[2,2]{1,0} multiply(s32[2,2]{1,0} %add, s32[2,2]{1,0} %iota)
ROOT %sub = s32[2,2]{1,0} subtract(s32[2,2]{1,0} %add, s32[2,2]{1,0} %iota)
}
// CHECK-NOT: store
// CHECK: %[[RESULT:.*]] = muli %{{.*}}, %{{.*}}
// CHECK: store %[[RESULT]]
// CHECK: [[RESULT:%.*]] = subi %{{.*}}, %{{.*}}
// CHECK: store [[RESULT]]