fix tests

google · Mar 29, 2024 · 5e067e5 · 5e067e5
1 parent 3d9974b
commit 5e067e5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 6 deletions.
diff --git a/tests/heir_simd_vectorizer/dot_product_8.mlir b/tests/heir_simd_vectorizer/dot_product_8.mlir
@@ -2,6 +2,8 @@
 // RUN:   --heir-simd-vectorizer %s | FileCheck %s
 
 // CHECK-LABEL: func @dot_product
+// CHECK-COUNT-3: tensor_ext.rotate
+// CHECK-NOT: tensor_ext.rotate
 func.func @dot_product(%arg0: tensor<8xi16>, %arg1: tensor<8xi16>) -> i16 {
   %c0 = arith.constant 0 : index
   %c0_si16 = arith.constant 0 : i16

diff --git a/tests/heir_simd_vectorizer/hamming_distance.mlir b/tests/heir_simd_vectorizer/hamming_distance.mlir
@@ -9,15 +9,9 @@
 // CHECK-NEXT: arith.addi
 // CHECK-NEXT: tensor_ext.rotate
 // CHECK-NEXT: arith.addi
-// CHECK-NEXT: tensor_ext.rotate
-// CHECK-NEXT: arith.addi
 // CHECK-NEXT: tensor.extract
 // CHECK-NEXT: secret.yield
 
-// TODO(#521): Fix rotate-and-reduce to work on this IR.
-// The problem is that the lattice identifies the rotate-version of this IR as
-// being overdetermined.
-
 func.func @hamming(%arg0: tensor<4xi16>, %arg1: tensor<4xi16>) -> i16 {
   %c0 = arith.constant 0 : index
   %c0_si16 = arith.constant 0 : i16