g;6dZddlZddlZddlZddlZddlmZddlmZddl m Z ddl m Z ddl mZddlmZdd lmZejeZejd d Z ejd d dZGddeZdZdZddZdZdZdZdZ dZ!dS)zAMap input circuit onto a backend topology via insertion of SWAPs.N)SwapGate)TransformationPass)TranspilerError)Layout) DAGOpNode)Target)disjoint_utils_Step)state swaps_added gates_mappedgates_remaining _SystemState)layout coupling_mapregisterswapsN)defaultsc*eZdZdZdfd ZdZxZS) LookaheadSwapaMap input circuit onto a backend topology via insertion of SWAPs. Implementation of Sven Jandura's swap mapper submission for the 2018 Qiskit Developer Challenge, adapted to integrate into the transpiler architecture. The role of the swapper pass is to modify the starting circuit to be compatible with the target device's topology (the set of two-qubit gates available on the hardware.) To do this, the pass will insert SWAP gates to relocate the virtual qubits for each upcoming gate onto a set of coupled physical qubits. However, as SWAP gates are particularly lossy, the goal is to accomplish this remapping while introducing the fewest possible additional SWAPs. This algorithm searches through the available combinations of SWAP gates by means of a narrowed best first/beam search, described as follows: - Start with a layout of virtual qubits onto physical qubits. - Find any gates in the input circuit which can be performed with the current layout and mark them as mapped. - For all possible SWAP gates, calculate the layout that would result from their application and rank them according to the distance of the resulting layout over upcoming gates (see _calc_layout_distance.) - For the four (search_width) highest-ranking SWAPs, repeat the above process on the layout that would be generated if they were applied. - Repeat this process down to a depth of four (search_depth) SWAPs away from the initial layout, for a total of 256 (search_width^search_depth) prospective layouts. - Choose the layout which maximizes the number of two-qubit which could be performed. Add its mapped gates, including the SWAPs generated, to the output circuit. - Repeat the above until all gates from the initial circuit are mapped. For more details on the algorithm, see Sven's blog post: https://medium.com/qiskit/improving-a-quantum-compiler-48410d7a7084 Fctt|tr&||_|j|_nd|_||_||_||_||_ dS)aLookaheadSwap initializer. Args: coupling_map (Union[CouplingMap, Target]): CouplingMap of the target backend. search_depth (int): lookahead tree depth when ranking best SWAP options. search_width (int): lookahead tree width when ranking best SWAP options. fake_run (bool): if true, it will only pretend to do routing, i.e., no swap is effectively added. N) super__init__ isinstancertargetbuild_coupling_mapr search_depth search_widthfake_run)selfrrr r! __class__s e/granary/edev/quantum/lib/python3.11/site-packages/qiskit/transpiler/passes/routing/lookahead_swap.pyrzLookaheadSwap.__init__Vsw  lF + + -&DK $ > > @ @D  DK ,D ((  c|jtdt|jdks|jddtdt|jj}t|j|kr(tdt|jd|dtj||j |jn|j |jd}ttj ||j|}g}t|}|rtd t|t#|||j|j}|td td t|j|j|j}|j}|j}||||jd |j|jd <n3|j|jd |j|jd <|jr|S|} |D]*} | | j| j | j!d +| S)aRun the LookaheadSwap pass on `dag`. Args: dag (DAGCircuit): the directed acyclic graph to be mapped Returns: DAGCircuit: A dag mapped to be compatible with the coupling_map in the property_set. Raises: TranspilerError: if the coupling map or the layout are not compatible with the DAG, or if the coupling_map=None Nz/LookaheadSwap cannot run with coupling_map=Noneqz-Lookahead swap runs on physical circuits onlyzThe number of DAG qubits (z9) is greater than the number of available device qubits (z).z+Top-level routing step: %d gates remaining.zLLookahead failed to find a swap which mapped gates or improved layout score.z2Found best step: mapped %d gates. Added swaps: %s. final_layoutF)check)"rrlenqregsgetphysical_qubitsqubitsr $require_layout_isolated_to_componentrrrgenerate_trivial_layoutlist serial_layersloggerdebug_search_forward_n_swapsrr r r r rextend property_setrcomposer!copy_empty_likeapply_operation_backopqargscargs) r"dagnumber_of_available_qubitsr current_state mapped_gatesr best_stepr mapped_dagnodes r$runzLookaheadSwap.runls   $!"STT T sy>>Q  #)--T":":"B!"QRR R%():)J%K%K" sz??7 7 7!KS__KK,FKKK  ; dk&9""t{   9S>$  *8 4 4d6G   s002233 . LLFOH\H\ ] ] ]/!! I %b LLDI*++%    &OM$1L'7O    - - -3 .6  ^ , 40=0DD n - -0=0D0L0L!.13:11D n - = J((**   Z ZD  + +DGTZSX + Y Y Y Yr%)rrF)__name__ __module__ __qualname____doc__rrF __classcell__)r#s@r$rr2s^!!F!!!!!!,OOOOOOOr%rc |j7djDt \}}t g||}|r|dkr|St fdjDd}td|d |d|d zDddtj } } }t|D]\} \} } }t|||d z |}|!t|}|| kr,td || g|jz|| ||} } }| t|t!|d z krm| kt!| j|ksQt!| jt!|ks,t'| j| jt'||krndSt+|}t | j|g| jz||z| jz| j}td ||j|S)aSearch for SWAPs which allow for application of largest number of gates. Args: state (_SystemState): The ``namedtuple`` collection containing the state of the physical system. This includes the current layout, the coupling map, the canonical register and the possible swaps available. gates (list): Gates to be mapped. depth (int): Number of SWAP layers to search before choosing a result. width (int): Number of SWAPs to consider at each layer. Returns: Optional(_Step): Describes the solution step found. If ``None``, no swaps leading to an improvement were found. Nc0h|]\}}||kr||fn||fSrN).0abs r$ z*_search_forward_n_swaps..s0\\\TQa!eeQFF!Q\\\r%)rrc3:K|]}t|VdSr)_score_state_with_swap)rOswapgatesr s r$ z*_search_forward_n_swaps..s0LL eU 3 3LLLLLLr%c|dS)NrrN)xs r$z)_search_forward_n_swaps..s adr%)keyz*At depth %d, ranked candidate swaps: %s...cg|] \}}}||f SrNrN)rOscorerU_s r$ z+_search_forward_n_swaps..s"GGG>5$$GGGr%r'z0At depth %d, updating best step: %s (score: %f).zAt depth %d, best_swap set: %s.)r_replacer get_edges_map_free_gatesr sortedr4r5mathinf enumerater6 _score_stepr minr+r r_calc_layout_distancer _swap_ops_from_edge)r rVdepthwidthr r base_step ranked_swaps best_swaprC best_scorerankr^rU new_state next_step next_scorebest_swap_gateouts`` r$r6r6s {\\U=O=Y=Y=[=[\\\  %4E5$A$A!L/eR??I eqjjLLLLL LLL NL LL4 GG\+EAI+-FGGG (,TDH9*yI&/ &=&=##""q$ +IPQ SXYY    ++  " " LLB..     04Y *yI Cs<001455 5 5%I*++e33y011C4H4HHH))*CY_UU+OYGGHH Et(E::N   i++~% (>>!   C  LL2E3?KKK Jr%ct}g}g}|jj}|D]}|dst|dj}|s(||r+||||n%t||}|||dd}||r+||||t|dkr't||}|||j ||d||ddkr't||}||t||||||fS)aMap all gates that can be executed with the current layout. Args: state (_SystemState): The physical characteristics of the system, including its current layout and the coupling map. gates (list): Gates to be mapped. Returns: tuple: mapped_gates (list): ops for gates that can be executed, mapped onto layout. remaining_gates (list): gates that cannot be executed on the layout. partitiongraphrr') setr_v2p_first_op_noder= intersectionupdateappend_transform_gate_for_systemr+rdistance) r rVblocked_qubitsrBremaining_gates layout_mapgater/ mapped_gates r$rcrcsUUNLO"J))K #DM228F **622 1%%f---&&t,,,,8uEE ##K000 k"1%  & &v . . )  ! !& ) ) )  " "4 ( ( ( ( [[A  4T5AAK    , , , ,   ( (F1I)> 6RS9@U V VZ[ [ [4T5AAK    , , , ,  ! !& ) ) )  " "4 ( ( ( (  ((r%c>|ddt|jjzz}|jj}d}|d|D]b}|ds |dd}t|dkr6||j||d||dz }c|S)z{Return the sum of the distances of two-qubit pairs in each CNOT in gates according to the layout and the coupling. N2 rryr`r')r+rr.rr|r)rVr max_gatesrrwrr/s r$rjrjHsc%"4"DEEEE "J Cjyj!]]K   k"1% v;;!   5%..z&)/DjQWXYQZF[\\ \C Jr%c|j}|j|||}t ||||fS)zCalculate the relative score for a given SWAP. Returns: float: the score of the given swap. Tuple[int, int]: the input swap that should be performed. _SystemState: an updated system state with the new layout contained. )r)rcopyrUrarj)rUr rV trial_layoutrss r$rTrTZsQ<$$&&LLtl33I  2 2D) CCr%cntd|jDdt|jzz S)zACount the mapped two-qubit gates, less the number of added SWAPs.cBg|]}t|jdk|S)r`)r+r=)rOgs r$r_z_score_step..ks*BBBaAG 0A0A0A0A0Ar%)r+r r )steps r$rhrhhs9 BB4,BBB C Ca#dN^J_J_F_ __r%ctjt|d}|j|jjt fd|jD|_|S)z6Return op implementing a virtual gate on given layout.rzc34K|]}|VdSrrN)rOrP device_qregrs r$rWz-_transform_gate_for_system..ts, Z ZZ]!; Z Z Z Z Z Zr%)rr}rrr|tupler=)rr mapped_op_noderrs @@r$rrnsaY~d7m<<==N.K"J Z Z Z Z Z^EY Z Z ZZZN r%c|jtfd|D}tt|dgS)zDGenerate list of ops to implement a SWAP gate along a coupling edge.c3(K|] }|V dSrrN)rOirs r$rWz&_swap_ops_from_edge..|s'33k!n333333r%rN)r<r=r>)rrrr)edger qreg_edgers @r$rkrkysJ.K3333d33333I 9B ? ? ? @@r%cXtd|DS)z!Get the first op node from a DAG.c3DK|]}t|t|VdSr)rr)rOrEs r$rWz!_first_op_node..s1LL 40K0KLLLLLLLr%)nextnodes)r?s r$r}r}s) LLLLL L LLr%)"rJ collectionsrloggingre%qiskit.circuit.library.standard_gatesrqiskit.transpiler.basepassesrqiskit.transpiler.exceptionsrqiskit.transpiler.layoutrqiskit.dagcircuitrqiskit.transpiler.targetrqiskit.transpiler.passes.layoutr getLoggerrGr4 namedtupler rrr6rcrjrTrhrrkr}rNr%r$rsHG  ::::::;;;;;;888888++++++''''''++++++::::::  8 $ $  w(cdd&{%3   IIIII&IIIXQQQh3)3)3)l$ D D D``` AAAMMMMMr%