Fix format in tests

Hatsunespica · Hatsunespica · commit 7711518c7098 · 2025-12-25T15:03:45.000-08:00
diff --git a/tests/filecheck/dialects/tensor-theory/ops.mlir b/tests/filecheck/dialects/tensor-theory/ops.mlir
@@ -16,5 +16,5 @@ builtin.module {
   %zero = "smt.fp.pzero"() : () -> !smt.fp<8,24>
   %eq_zero = "smt.eq"(%extract, %zero) : (!smt.fp<8,24>, !smt.fp<8,24>) -> !smt.bool
   "smt.assert"(%eq_zero) : (!smt.bool) -> ()
-  // CHECK-NEXT: // CHECK-NEXT: (declare-const $idx2 (_ BitVec 64))
+  // CHECK-NEXT: (assert (= (select (select $tmp $idx1) $idx2) (_ +zero 8 24)))
 }
diff --git a/tests/filecheck/rewrite-tensors/rewrite-smt-tensor.mlir b/tests/filecheck/rewrite-tensors/rewrite-smt-tensor.mlir
@@ -11,3 +11,8 @@ builtin.module {
   %extract = "smt.tensor.extract"(%transpose, %idx1, %idx2):
         (!smt.tensor.tensor<[3, 3], !smt.fp<8, 24>, none>, !smt.bv<64>, !smt.bv<64>) -> !smt.fp<8, 24>
 }
+
+// CHECK:       %0 = "smt.declare_const"() : () -> !smt.tensor.tensor<[3 : i64, 3 : i64], !smt.fp<8, 24>, none>
+// CHECK-NEXT:  %idx1 = "smt.declare_const"() : () -> !smt.bv<64>
+// CHECK-NEXT:  %idx2 = "smt.declare_const"() : () -> !smt.bv<64>
+// CHECK-NEXT:  %extract = "smt.tensor.extract"(%0, %idx2, %idx1) : (!smt.tensor.tensor<[3 : i64, 3 : i64], !smt.fp<8, 24>, none>, !smt.bv<64>, !smt.bv<64>) -> !smt.fp<8, 24>

Original file line number	Diff line number	Diff line change
`@@ -16,5 +16,5 @@ builtin.module {`
`16`	`16`	`%zero = "smt.fp.pzero"() : () -> !smt.fp<8,24>`
`17`	`17`	`%eq_zero = "smt.eq"(%extract, %zero) : (!smt.fp<8,24>, !smt.fp<8,24>) -> !smt.bool`
`18`	`18`	`"smt.assert"(%eq_zero) : (!smt.bool) -> ()`
`19`		`- // CHECK-NEXT: // CHECK-NEXT: (declare-const $idx2 (_ BitVec 64))`
	`19`	`+ // CHECK-NEXT: (assert (= (select (select $tmp $idx1) $idx2) (_ +zero 8 24)))`
`20`	`20`	`}`
Original file line number	Diff line number	Diff line change
`@@ -11,3 +11,8 @@ builtin.module {`
`11`	`11`	`%extract = "smt.tensor.extract"(%transpose, %idx1, %idx2):`
`12`	`12`	`(!smt.tensor.tensor<[3, 3], !smt.fp<8, 24>, none>, !smt.bv<64>, !smt.bv<64>) -> !smt.fp<8, 24>`
`13`	`13`	`}`
	`14`	`+`
	`15`	`+// CHECK: %0 = "smt.declare_const"() : () -> !smt.tensor.tensor<[3 : i64, 3 : i64], !smt.fp<8, 24>, none>`
	`16`	`+// CHECK-NEXT: %idx1 = "smt.declare_const"() : () -> !smt.bv<64>`
	`17`	`+// CHECK-NEXT: %idx2 = "smt.declare_const"() : () -> !smt.bv<64>`
	`18`	`+// CHECK-NEXT: %extract = "smt.tensor.extract"(%0, %idx2, %idx1) : (!smt.tensor.tensor<[3 : i64, 3 : i64], !smt.fp<8, 24>, none>, !smt.bv<64>, !smt.bv<64>) -> !smt.fp<8, 24>`