Verify

PennyLaneAI · Aug 13, 2024 · deeaf15 · deeaf15
1 parent 0ddb868
commit deeaf15
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/frontend/catalyst/device/verification.py b/frontend/catalyst/device/verification.py
@@ -212,7 +212,9 @@ def _op_checker(op, state):
         if type(op) in (Controlled, ControlledOp) or isinstance(op, (Adjoint)):
             pass
         # Don't check StatePrep since StatePrep is not in the list of device capabilities.
-        elif isinstance(op, StatePrepBase):
+        elif (
+            isinstance(op, StatePrepBase) and qjit_device.qjit_capabilities.initial_state_prep_flag
+        ):
             pass
         elif not qjit_device.qjit_capabilities.native_ops.get(op.name):
             raise CompileError(