Fix python check lines.

Signed-off-by: Eric Schweitz <[email protected]>
NVIDIA · Nov 23, 2024 · 3e29b19 · 3e29b19
1 parent a03a5ef
commit 3e29b19
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/python/tests/mlir/ast_compute_action.py b/python/tests/mlir/ast_compute_action.py
@@ -34,7 +34,7 @@ def compute():
 # CHECK-DAG:           %[[VAL_3:.*]] = arith.constant 1 : i64
 # CHECK:           %[[VAL_4:.*]] = quake.veq_size %[[VAL_0]] : (!quake.veq<?>) -> i64
 # CHECK:           %[[VAL_5:.*]] = arith.subi %[[VAL_4]], %[[VAL_1]] : i64
-# CHECK:           %[[VAL_6:.*]] = quake.subveq %[[VAL_0]], %[[VAL_2]], %[[VAL_5]] : (!quake.veq<?>, i64, i64) -> !quake.veq<?>
+# CHECK:           %[[VAL_6:.*]] = quake.subveq %[[VAL_0]], 0, %[[VAL_5]] : (!quake.veq<?>, i64) -> !quake.veq<?>
 # CHECK:           %[[VAL_7:.*]] = arith.subi %[[VAL_4]], %[[VAL_3]] : i64
 # CHECK:           %[[VAL_8:.*]] = quake.extract_ref %[[VAL_0]]{{\[}}%[[VAL_7]]] : (!quake.veq<?>, i64) -> !quake.ref
 # CHECK:           %[[VAL_9:.*]] = cc.create_lambda {

diff --git a/python/tests/mlir/ast_lambda_tuple_stmts.py b/python/tests/mlir/ast_lambda_tuple_stmts.py
@@ -30,7 +30,7 @@ def reflect(qubits: cudaq.qview):
 # CHECK-DAG:           %[[VAL_3:.*]] = arith.constant 1 : i64
 # CHECK:           %[[VAL_4:.*]] = quake.veq_size %[[VAL_0]] : (!quake.veq<?>) -> i64
 # CHECK:           %[[VAL_5:.*]] = arith.subi %[[VAL_4]], %[[VAL_1]] : i64
-# CHECK:           %[[VAL_6:.*]] = quake.subveq %[[VAL_0]], %[[VAL_2]], %[[VAL_5]] : (!quake.veq<?>, i64, i64) -> !quake.veq<?>
+# CHECK:           %[[VAL_6:.*]] = quake.subveq %[[VAL_0]], 0, %[[VAL_5]] : (!quake.veq<?>, i64) -> !quake.veq<?>
 # CHECK:           %[[VAL_7:.*]] = arith.subi %[[VAL_4]], %[[VAL_3]] : i64
 # CHECK:           %[[VAL_8:.*]] = quake.extract_ref %[[VAL_0]]{{\[}}%[[VAL_7]]] : (!quake.veq<?>, i64) -> !quake.ref
 # CHECK:           %[[VAL_9:.*]] = cc.create_lambda {

diff --git a/python/tests/mlir/ast_qreg_slice.py b/python/tests/mlir/ast_qreg_slice.py
@@ -59,7 +59,7 @@ def slice():
 # CHECK-DAG:           %[[VAL_5:.*]] = arith.constant 5 : i64
 # CHECK-DAG:           %[[VAL_6:.*]] = arith.constant 0 : i64
 # CHECK:           %[[VAL_7:.*]] = quake.alloca !quake.veq<4>
-# CHECK:           %[[VAL_8:.*]] = quake.subveq %[[VAL_7]], %[[VAL_2]], %[[VAL_0]] : (!quake.veq<4>, i64, i64) -> !quake.veq<2>
+# CHECK:           %[[VAL_8:.*]] = quake.subveq %[[VAL_7]], 2, 3 : (!quake.veq<4>) -> !quake.veq<2>
 # CHECK:           %[[VAL_9:.*]] = quake.extract_ref %[[VAL_8]][0] : (!quake.veq<2>) -> !quake.ref
 # CHECK:           quake.x %[[VAL_9]] : (!quake.ref) -> ()
 # CHECK:           %[[VAL_10:.*]] = quake.extract_ref %[[VAL_8]][1] : (!quake.veq<2>) -> !quake.ref

diff --git a/python/tests/mlir/ghz.py b/python/tests/mlir/ghz.py
@@ -75,7 +75,7 @@ def simple(numQubits: int):
 # CHECK:           quake.h %[[VAL_7]] : (!quake.ref) -> ()
 # CHECK:           %[[VAL_8:.*]] = cc.load %[[VAL_4]] : !cc.ptr<i64>
 # CHECK:           %[[VAL_9:.*]] = arith.subi %[[VAL_8]], %[[VAL_1]] : i64
-# CHECK:           %[[VAL_10:.*]] = quake.subveq %[[VAL_6]], %[[VAL_3]], %[[VAL_9]] : (!quake.veq<?>, i64, i64) -> !quake.veq<?>
+# CHECK:           %[[VAL_10:.*]] = quake.subveq %[[VAL_6]], 0, %[[VAL_9]] : (!quake.veq<?>, i64) -> !quake.veq<?>
 # CHECK:           %[[VAL_11:.*]] = quake.veq_size %[[VAL_10]] : (!quake.veq<?>) -> i64
 # CHECK:           %[[VAL_12:.*]] = cc.alloca !cc.struct<{i64, !quake.ref}>{{\[}}%[[VAL_11]] : i64]
 # CHECK:           %[[VAL_13:.*]] = cc.loop while ((%[[VAL_14:.*]] = %[[VAL_3]]) -> (i64)) {