update tests

google · Mar 8, 2024 · 0adf17d · 0adf17d
1 parent ff64893
commit 0adf17d
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 909 deletions.
diff --git a/tests/simd/box_blur_4x4.mlir b/tests/simd/box_blur_4x4.mlir
@@ -1,10 +1,12 @@
 // RUN: heir-opt --secretize=entry-function=box_blur --wrap-generic --canonicalize --cse \
 // RUN:   --full-loop-unroll \
-// RUN:   --insert-rotate \
+// RUN:   --insert-rotate --cse --canonicalize --collapse-insertion-chains --canonicalize --cse \
 // RUN:   %s | FileCheck %s
 
 module  {
   // CHECK-LABEL: @box_blur
+  // CHECK-NOT: tensor.extract
+  // CHECK-COUNT-7: tensor_ext.rotate
   func.func @box_blur(%arg0: tensor<16xi16> {secret.secret}) -> tensor<16xi16> {
     %c16 = arith.constant 16 : index
     %c4 = arith.constant 4 : index

diff --git a/tests/simd/box_blur_4x4_unrolled.mlir b/tests/simd/box_blur_4x4_unrolled.mlir
diff --git a/tests/simd/box_blur_64x64.mlir b/tests/simd/box_blur_64x64.mlir
@@ -1,10 +1,12 @@
 // RUN: heir-opt --secretize=entry-function=box_blur --wrap-generic --canonicalize --cse \
-// RUN:   --affine-loop-unroll=unroll-factor=2 \
-// RUN:   --affine-loop-unroll=unroll-factor=2 --canonicalize --cse \
-// RUN:   --affine-loop-unroll=unroll-factor=64 --canonicalize --cse \
-// RUN:   < %s
+// RUN:   --full-loop-unroll \
+// RUN:   --insert-rotate --cse --canonicalize --collapse-insertion-chains --canonicalize --cse \
+// RUN:   %s | FileCheck %s
 
 module  {
+  // CHECK-LABEL: @box_blur
+  // CHECK-NOT: tensor.extract
+  // CHECK-COUNT-7: tensor_ext.rotate
   func.func @box_blur(%arg0: tensor<4096xi16> {secret.secret}) -> tensor<4096xi16> {
     %c4096 = arith.constant 4096 : index
     %c64 = arith.constant 64 : index