commit 8b2004625cc12740cebed57ea15cece0ff42c684
author Lei Zhang <antiagainst@google.com> Mon Jun 07 12:59:36 2021 -0400
committer GitHub <noreply@github.com> Tue Jun 08 00:59:36 2021 +0800
tree e4f7eab3d4c0a20d1fb54b6c1d1d57ea330db13b
parent 279796db5743e3f4104215319d8fcb42e1c58145

Disable large_dot_general2 test case util it gets fixed (#6129)

Right now we don't have a good way to disable per GPU vendor,
but that would be nice to have. For now just disable for all
to avoid marking all commits as failing.

iree/test/e2e/vulkan_specific/dot_general.mlir

diff --git a/iree/test/e2e/vulkan_specific/dot_general.mlir b/iree/test/e2e/vulkan_specific/dot_general.mlir
index 11cd69d..250afb9 100644
--- a/iree/test/e2e/vulkan_specific/dot_general.mlir
+++ b/iree/test/e2e/vulkan_specific/dot_general.mlir
@@ -48,18 +48,19 @@
   return
 }
 
-func @large_dot_general2() attributes { iree.module.export } {
-  %lhs = iree.unfoldable_constant dense<1.0> : tensor<4x32x1024xf32>
-  %rhs = iree.unfoldable_constant dense<0.4> : tensor<4x1024x64xf32>
-  %res = "mhlo.dot_general"(%lhs, %rhs) {
-           dot_dimension_numbers = {
-             lhs_batching_dimensions = dense<0> : tensor<1xi64>,
-             lhs_contracting_dimensions = dense<2> : tensor<1xi64>,
-             rhs_batching_dimensions = dense<0> : tensor<1xi64>,
-             rhs_contracting_dimensions = dense<1> : tensor<1xi64>
-           },
-           precision_config = ["DEFAULT", "DEFAULT"]
-         } : (tensor<4x32x1024xf32>, tensor<4x1024x64xf32>) -> tensor<4x32x64xf32>
-  check.expect_almost_eq_const(%res, dense<409.596> : tensor<4x32x64xf32>) : tensor<4x32x64xf32>
-  return
-}
+// TODO(GH-6070): Re-enable this after fixing wrong answer on Pixel 4.
+// func @large_dot_general2() attributes { iree.module.export } {
+//   %lhs = iree.unfoldable_constant dense<1.0> : tensor<4x32x1024xf32>
+//   %rhs = iree.unfoldable_constant dense<0.4> : tensor<4x1024x64xf32>
+//   %res = "mhlo.dot_general"(%lhs, %rhs) {
+//            dot_dimension_numbers = {
+//              lhs_batching_dimensions = dense<0> : tensor<1xi64>,
+//              lhs_contracting_dimensions = dense<2> : tensor<1xi64>,
+//              rhs_batching_dimensions = dense<0> : tensor<1xi64>,
+//              rhs_contracting_dimensions = dense<1> : tensor<1xi64>
+//            },
+//            precision_config = ["DEFAULT", "DEFAULT"]
+//          } : (tensor<4x32x1024xf32>, tensor<4x1024x64xf32>) -> tensor<4x32x64xf32>
+//   check.expect_almost_eq_const(%res, dense<409.596> : tensor<4x32x64xf32>) : tensor<4x32x64xf32>
+//   return
+// }