z i?SrSSKJr SSKJrJrJr SSKJr SSK J r SSK J r SSK Jr SSKJrJrJr SS KJr \R,R."S S \55rg ) z*Pass for Hoare logic circuit optimization.)TransformationPass)QuantumRegisterControlledGateGate) DAGCircuit) UnitaryGate) matrix_equal) CircuitError)CZGateCU1GateMCU1Gate) optionalsc^\rSrSrSrSU4SjjrSrSrSrSr Sr S r S r SS jr S rS rSSjrSrSrSrSrU=r$)HoareOptimizerzThis is a transpiler pass using Hoare logic circuit optimization. The inner workings of this are detailed in: https://arxiv.org/abs/1810.00375 c>SSKJn [TU] 5 SUlSUlSUlSUlSUlXl g)z Args: size (int): size of gate cache, in number of gates Raises: MissingOptionalLibraryError: if unable to import z3 solver )_gate_extensionN) rsuper__init__solver variablesgatenum gatecachevarnumsize)selfrr __class__s i/home/james-whalen/.local/lib/python3.13/site-packages/qiskit/transpiler/passes/optimization/hoare_opt.pyrHoareOptimizer.__init__s; &     cSSKnS[U5-S-[URU5-nURU5nURU==S- ss'URUR U5 U$)a8After each gate generate a new unique variable name for each of the qubits, using scheme: 'q[id]_[gatenum]', e.g. q1_0 -> q1_1 -> q1_2, q2_0 -> q2_1 Args: qubit (Qubit): qubit to generate new variable for Returns: BoolRef: z3 variable of qubit state rNq_r)z3strrBoolrappend)rqubitr&varnamevars r _gen_variableHoareOptimizer._gen_variable1sk E "S(3t||E/B+CCggg Uq  u$$S) r"c SSKnURHznSURU'/URU'/URU'0UR U'UR U5nURRURU55 M| g)z}create boolean variables for each qubit and apply qb == 0 condition Args: dag (DAGCircuit): input DAG to get qubits from rN) r&qubitsrrrrr-raddNot)rdagr&qbtxs r _initializeHoareOptimizer._initializeBsu ::C !DLL "$DNN3 "$DNN3 !DKK ""3'A KKOOBFF1I & r"c SSKn/nUH#nURURU55 M% URR UR X!R "XF-655 [U5HDupURR UR URU5XhU :H55 MF g![a N`f=f)acreate boolean variables for each qubit the gate is applied to and apply the relevant post conditions. a gate rotating out of the z-basis will not have any valid post-conditions, in which case the qubit state is unknown Args: gate (Gate): gate to inspect ctrl_ones (BoolRef): z3 condition asserting all control qubits to 1 trgtqb (list((QuantumRegister, int))): list of target qubits trgtvar (list(BoolRef)): z3 variables corresponding to latest state of target qubits rN) r&r)r-rr1Implies_postconditionsAttributeError enumerater2) rgate ctrl_onestrgtqbtrgtvarr&new_varsr4itvars r _add_postconditions"HoareOptimizer._add_postconditionsQs C OOD..s3 4  KKOOBJJy2F2FI[2]^ _!)GA KKOOBJJrvvi'8(+:MN O*   s9B>> C  C c HSSKnSnURR5 UR"U6n[ U[ 5(GaU(Ga[ U5S:XaURRURURX#S555 URR5UR:HnURR5 URR5 URRURX#S55 URR5UR:HnU=(d UnOaURRURX$RU555 URR5UR:HnURR5 U$![aE URRU5 URR5UR:HnNjf=f)a=use z3 sat solver to determine triviality of gate Args: gate (Gate): gate to inspect ctrl_ones (BoolRef): z3 condition asserting all control qubits to 1 trgtvar (list(BoolRef)): z3 variables corresponding to latest state of target qubits Returns: bool: if gate is trivial rNFr)r&rpush _trivial_if isinstanceboollenr1r2Andcheckunsatpopr;) rr=r>r@r&trivial triv_condsol1sol2s r _test_gateHoareOptimizer._test_gateks{   :(('2I )T**W!2KKOOBFF266)QZ+H$IJ;;,,."((:DKKOO%KK$$&KKOOBFF9aj$AB;;,,."((:D"ldG y&&2C DE++++-9 % 6 KKOOI &kk'')RXX5G 6sGA H! H!cjSn[UR5n[5nURU5 [ [ [ U5UR55n[U[5(aURU5n[U[[[45(aTU(dMUSS:aDUSS- US'USUSX'SS-S-U-nURU5nU(d USS:aMDU(a0UV s/sHoU PM n n URURU 5 XFU4$s sn f)aMuse z3 sat solver to determine if all control qubits are in 1 state, and if so replace the Controlled - U by U. Args: gate (Gate): gate to inspect ctrlvar (list(BoolRef)): z3 variables corresponding to latest state of control qubits trgtvar (list(BoolRef)): z3 variables corresponding to latest state of target qubits Returns: Tuple(bool, DAGCircuit, List):: * bool:if controlled gate can be replaced. * DAGCircuit: with U applied to the target qubits. * List: with indices of target qubits. FrrN)r num_qubitsradd_qreglistrangerKrIr_check_removalr r r apply_operation_back base_gate) rr=ctrlvarr@removeqargr3qb ctrl_varsqir0s r _remove_controlHoareOptimizer._remove_controlst/l T %G doo6 7 dN + +((1F dVWh7 8 8A1 1#Gbe,w!uqy{/CCgM ,,Y7A )+,22hF,  $ $T^^V <B-sD0c6SSKnUR"U6nURR5 URR UR U55 URR 5UR:HnURR5 U$)Nr) r&rLrrGr1r2rMrNrO)rr^r&r>r_s r r[HoareOptimizer._check_removalsjFFG$   y)*""$0  r"cSSKn[UR55nUGHnURnUR U5upgpUR "U6n UR XWU 5upnU (abURXL5n [[U R555nURnUR U5upgpUR "U6n URXZU 5nU(aURU5 OURS:aURHCnURUR!U5 UR"US- UR$UU'ME URH<n['URU5UR:dM+UR)X5 M> UR+XZX5 GM g)atraverse DAG in topological order for each gate check: if any control is 0, or if triviality conditions are satisfied if yes remove gate from dag apply postconditions of gate Args: dag (DAGCircuit): input DAG to optimize in place rNr)r&rYtopological_op_nodesop_seperate_ctrl_trgtrLrdsubstitute_node_with_dagnextitervaluesrTremove_op_noderqargsrr)rrrK_multigate_optrD)rr3r&nodesnoder=r%r^r?r@r> remove_ctrlnew_dag mapped_nodesrPr4s r _traverse_dagHoareOptimizer._traverse_dags~ S--/0D77D*.*B*B4*H 'A(I&*&:&:4'&R #K! #;;DJ D!4!4!678ww.2.F.Ft.L+FFFG, oodw?G""4(Q::CNN3'..t4-1\\#->-BDKK$T*& ::C4>>#./499<++C5&  $ $Tf F=r"cZSnU[URU5S- :GaqSnURUUnURUUS-nURU5SnURU5Sn US- nX:waM}UH=n URU RU5n URU U S-ULdM;SnM? Xg/n U(aUR U 5(aURU 5(awUS- nU Hln URU 5 Ub$URURU 5U:dM;U RH!n URU RU 5 M# Mn U[URU5S- :aGMpgg![[ 4a GMf=f)aremove gates that have the same set of target qubits, follow each other immediately on these target qubits, and combine to the identity (consider sequences of length 2 for now) Args: dag (DAGCircuit): the directed acyclic graph to run on. qubit (Qubit): qubit cache to inspect from_idx (int): only gates whose indexes in the cache are larger than this value can be removed rrTFN) rKrrkindex IndexError ValueError _is_identity _seq_as_onerprqr_)rr3r*from_idxrBr)node1node2trgtqb1trgtqb2r4idxseqrts r _remove_successive_identity*HoareOptimizer._remove_successive_identitys #dnnU+,q00FNN5)!,ENN5)!a%0E..u5a8G..u5a8G FA! "C..-33E:C~~c*3735@!&#.C$++C00T5E5Ec5J5JQD&&t,'4>>%+@+F+Ft+Lx+W#'::C NN3/66t<$. +#dnnU+,q000 +  s9F>FF*)F*c`[U5S:XdeUSRUSRR5p2URURpTUR UR pv[ U[5(a URnURn[ U[5(a URnURnU[LaU[LaXg:H=(a U=(a U$U[LaU[La[USUSSS9$X#:H=(a XE:H$![a gf=f)zdetermine whether the sequence of gates combines to the identity (consider sequences of length 2 for now) Args: sequence (list(DAGOpNode)): gate sequence to inspect Returns: bool: if gate sequence combines to identity r{rrFT) ignore_phase) rKrjinverser params definitionrIrr] base_classrrr )rsequencegate1gate2par1par2def1def2s r rHoareOptimizer._is_identitys 8}!!! #A;>>8A;>>+A+A+C5\\5<3sH~#67Sh=PRUW_R`9a b kk!RXX-  r"c nURU(dgURXU5 [URU5UR:aUcgUc1Sn[ 5nUR U5 URUS/nOURUUSS2nUHnnUR Vs/sH owU;dM UPM nnURU5 UH1n URU RU5n URXXS9 M3 Mp URUUS-SURU'gs snf)aG Args: dag (DAGCircuit): the directed acyclic graph to run on. qubit (Qubit): qubit whose gate cache is to be optimized max_idx (int): a value indicates a recursive call, optimize and remove gates up to this point in the cache dnt_rec (list(int)): don't recurse on these qubit caches (again) Nr)max_idxdnt_recr) rrrKrsetr1rqupdater|rr) rr3r*rr gates_tbrrtr5new_qbr4rs r rrHoareOptimizer._multigate_optRs'~~e$  ((W= t~~e$ % 1go  _GeG KK .q12Iu-gkrk:ID!%@A/?aF@ NN6 "nnS)//5##Cc#K !%u 5gkm DuAs . D2;D2c<URn[U[5(a URnOSnURSUnURUSnUVs/sH$o`R UUR UUPM& nnUVs/sH$o`R UUR UUPM& nnXGXX4$s snfs snf![aQ UVs/sHo`R USPM Os snfnnUVs/sHo`R USPM Os snfnnNhf=f)zaGet the target qubits and control qubits if available, as well as their respective z3 variables. rNr)rjrIrnum_ctrl_qubitsrqrrKeyError) rrtr=numctrlctrlqbr?rar^r@s r rk"HoareOptimizer._seperate_ctrl_trgtxsww dN + +**GGHW%GH% @KQR6R~~b)$++b/$*?@6GRKQR6R~~b)$++b/$*?@6GR11 SR @8>?"~~b)"-?G?8>?"~~b)"-?G?G @sHC+B6?C+B;0C6 CDC*) D5DDDcnSSKnUR5Ul0Ul0Ul0Ul0Ulg)zEReset HoareOptimize internal state, so it can be run multiple times. rN)r&Solverrrrrr)rr&s r _resetHoareOptimizer._resets0 iik   r"cUR5 URU5 URU5 URS:a$URHnUR X5 M U$)zm Args: dag (DAGCircuit): the directed acyclic graph to run on. Returns: DAGCircuit: Transformed DAG. r)rr6rxrr0rr)rr3r4s r runHoareOptimizer.runsU   3 99q=zz##C-" r")rrrrrr) )N)NN)__name__ __module__ __qualname____firstlineno____doc__rr-r6rDrTrdr[rxrrrrrrkrr__static_attributes__ __classcell__)rs@r rrsa $" 'P4#J%N ,G\$=L /D2$EL2&   r"rN)rqiskit.transpiler.basepassesrqiskit.circuitrrrqiskit.dagcircuitrqiskit.circuit.libraryr(qiskit.quantum_info.operators.predicatesr qiskit.circuit.exceptionsr %qiskit.circuit.library.standard_gatesr r r qiskit.utilsr _optionalsHAS_Z3require_in_instancerr"r rsS1;@@(.A2KK0 &&K'K'Kr"