gD`dZddlmZddlZddlZddlZddlmZddlmZm Z ddl Z ddl m ZddlmZddlmZdd lmZmZmZmZmZmZmZdd lmZdd lmZdd lmZdd l m!Z!ddl"m#Z#ddl$m%Z%ddl&m'Z'Gddej(Z)Gdde%Z*Gdde*Z+dZ,dZ-d!d Z.dS)"zRZX calibration builders.) annotationsN)Sequence)pierf) Instruction)RZXGate) QiskitError)PlaySchedule ScheduleBlockControlChannel DriveChannelGaussianSquareWaveform)builder)filter_instructions)InstructionScheduleMap)Target)deprecate_pulse_dependency)CalibrationBuilder)CalibrationNotAvailablec*eZdZdZdZdZdZdZdZdZ dS) CRCalTypezAEstimated calibration type of backend cross resonance operations.z8Echoed Cross Resonance corresponding to native operationz2Echoed Cross Resonance reverse of native operationz;Echoed Cross Resonance CX corresponding to native operationz5Echoed Cross Resonance CX reverse of native operationz+Direct CX corresponding to native operationz%Direct CX reverse of native operationN) __name__ __module__ __qualname____doc__ ECR_FORWARD ECR_REVERSEECR_CX_FORWARDECR_CX_REVERSEDIRECT_CX_FORWARDDIRECT_CX_REVERSEf/granary/edev/quantum/lib/python3.11/site-packages/qiskit/transpiler/passes/calibration/rzx_builder.pyrr-s6KKLKFKRNLNE?r&rceZdZdZe ddfd ZddZeej dddZ d dZ xZ S)!RZXCalibrationBuildera Creates calibrations for RZXGate(theta) by stretching and compressing Gaussian square pulses in the CX gate. This is done by retrieving (for a given pair of qubits) the CX schedule in the instruction schedule map of the backend defaults. The CX schedule must be an echoed cross-resonance gate optionally with rotary tones. The cross-resonance drive tones and rotary pulses must be Gaussian square pulses. The width of the Gaussian square pulse is adjusted so as to match the desired rotation angle. If the rotation angle is small such that the width disappears then the amplitude of the zero width Gaussian square pulse (i.e. a Gaussian) is reduced to reach the target rotation angle. Additional details can be found in https://arxiv.org/abs/2012.11660. NTinstruction_schedule_maprverbosebooltargetrct||_||_|r||_|jt ddS)a Initializes a RZXGate calibration builder. Args: instruction_schedule_map: The :obj:`InstructionScheduleMap` object representing the default pulse calibrations for the target backend verbose: Set True to raise a user warning when RZX schedule cannot be built. target: The :class:`~.Target` representing the target backend, if both ``instruction_schedule_map`` and this are specified then this argument will take precedence and ``instruction_schedule_map`` will be ignored. Raises: QiskitError: Instruction schedule map is not provided. Nz8Calibrations can only be added to Pulse-enabled backends)super__init__ _inst_map_verbose_instruction_schedule_mapr )selfr*r+r- __class__s r'r0zRZXCalibrationBuilder.__init__Esd* 1  @#==??DN > !XYY Y " !r&node_op CircuitInstqubitslistreturncdt|tod|jjvp d|jjvS)zDetermine if a given node supports the calibration. Args: node_op: Target instruction object. qubits: Integer qubit indices to check. Returns: Return ``True`` is calibration can be provided. cxecr) isinstancerr1 instructions)r4r6r8s r' supportedzRZXCalibrationBuilder.supportedbs7'7++ DN/ / W5DNW>WW 7Om3 F %jj9,y8 -/ >>E 5MMMRWU^^ +MMMME 5MMMRWU^^k9MI IMMM 5/&/AA[P Q QT_ _ ,zw(22622 _k&9:::s 1,1Schedule | ScheduleBlockc |jd} t|}n"#t$r}td|d}~wwxYwt j|drt dSt|j|\}}}|tj tj fvr4|j r&tjd|d|jd t t"|tjtjfvrtj5tjd t, |jd |d}t1jd d|dd5} t5||D]p\} } t1j5|| ||| |dddn #1swxYwYt1j|q dddn #1swxYwYdddn #1swxYwY| S|jd |d}|jd|dt<dz } |jd|d} |jd|dt<dz }|jd|d}t1jd5}t1j| dt1j| dt1j| dt1j|dt1j|dt1j|ddddn #1swxYwYt1jd d|dd5}t1j|dt5||D]p\} } t1j5|| ||| |dddn #1swxYwYt1j|qt1j|ddddn #1swxYwY|S)aBuilds the calibration schedule for the RZXGate(theta) with echos. Args: node_op: Instruction of the RZXGate(theta). I.e. params[0] is theta. qubits: List of qubits for which to get the schedules. The first qubit is the control and the second is the target. Returns: schedule: The calibration schedule for the RZXGate(theta). Raises: QiskitError: if rotation angle is not assigned. QiskitError: If the control and target qubits cannot be identified. CalibrationNotAvailable: RZX schedule cannot be built for input node. rrHN rzx(0.000)nameCR instruction for qubits is likely ~ sequence. Pulse stretch for this calibration is not currently implemented. RZX schedule is not generated for this qubit pair.ignoreactioncategoryx sequentialrzx(.3f)default_alignmentrjrrzrLsxhadamardszcsxcsztsxt)r[rDrNr rUiscloser _check_calibration_typer1rr#r$r2warningswarnvalue UserWarningrr!rcatch_warnings simplefilterDeprecationWarninggetrbuildzip align_leftrdcallr)r4r6r8rCrZcal_typecr_tones comp_tonesxgaterzx_theta_nativecr_tone comp_toner|r}r~rr{rzx_theta_flips r'get_calibrationz%RZXCalibrationBuilder.get_calibrations q! P%LLEE P P PFGGR O P :eS ! ! 4 l333 3)@QW)X)X&(J  3Y5PQ Q Q}  IIIHNIII  * )  0)2GH H H(** , ,%X@RSSSS**3q ::]&29L9L9L9L9L,%.1(J.G.G,,*$/11CC 00%@@@ 00EBBBCCCCCCCCCCCCCCC U++++ ,,,,,,,,,,,,,,,, , , , , , , , , , , , , , , ,$ #""3q 22n  vay"q&99n  vay11n  vay"q&99n  vay11 ] + + + *x L5 ) ) ) ) L5 ) ) ) ) L5 ) ) ) ) L5 ) ) ) ) L5 ) ) ) ) L5 ) ) ) ) * * * * * * * * * * * * * * *]*1D1D1D1D1D   4  L 3 3 3 3&)(J&?&? $ $"'));;((%888((E:::;;;;;;;;;;;;;;; U#### L 3 3 3 3 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4s >9>;AG9)G"<-F5 ) G"5F9 9G"<F9 =G" G9"G& &G9)G& *G99G=G=BM##M'*M' ?Q-P5 QP QP .QQ Q)NTN)r*rr+r,r-r)r6r7r8r9r:r,)rA)rBr rCrDrErFr:rFr6r7r8r9r:re) rrrrrr0r@ staticmethodrmacrordr __classcell__)r5s@r'r)r)8s   <@ ZZZZZZ Z8      ]3333]\3jRRRRRRRRr&r)ceZdZdZd dZd S) RZXCalibrationBuilderNoEchoaS Creates calibrations for RZXGate(theta) by stretching and compressing Gaussian square pulses in the CX gate. The ``RZXCalibrationBuilderNoEcho`` is a variation of the :class:`~qiskit.transpiler.passes.RZXCalibrationBuilder` pass that creates calibrations for the cross-resonance pulses without inserting the echo pulses in the pulse schedule. This enables exposing the echo in the cross-resonance sequence as gates so that the transpiler can simplify them. The ``RZXCalibrationBuilderNoEcho`` only supports the hardware-native direction of the CX gate. r6r7r8r9r:rec |jd} t|}n"#t$r}td|d}~wwxYwt j|drt dSt|j|\}}}|tj tj fvr4|j r&tjd|d|jd t t"|tjtjfvrtj5tjd t, t/jd d |dd5}||dd|z} ||dd|zt/j| t7|ddddn #1swxYwYdddn #1swxYwY|Std)aBuilds the calibration schedule for the RZXGate(theta) without echos. Args: node_op: Instruction of the RZXGate(theta). I.e. params[0] is theta. qubits: List of qubits for which to get the schedules. The first qubit is the control and the second is the target. Returns: schedule: The calibration schedule for the RZXGate(theta). Raises: QiskitError: if rotation angle is not assigned. QiskitError: If the control and target qubits cannot be identified, or the backend does not natively support the specified direction of the cx. CalibrationNotAvailable: RZX schedule cannot be built for input node. rrHNrgrhrirkrlrmrnroleftrtrurvrwrLzDRZXCalibrationBuilderNoEcho only supports hardware-native RZX gates.)r[rDrNr rUrr rr1rr#r$r2rrrrrr!rrrrrrrddelayr) r4r6r8rCrZrrr rzx_theta stretched_durs r'rz+RZXCalibrationBuilderNoEcho.get_calibration s"q! P%LLEE P P PFGGR O P :eS ! ! 4 l333 3)@QW)X)X&(J  3Y5PQ Q Q}  IIIHNIII  * )  0)2GH H H(** J J%X@RSSSS]VBUBUBUBUBUVVVJZc$($8$8!a%i$P$PM((AE BBBM-fQi1H1HIII JJJJJJJJJJJJJJJ  J J J J J J J J J J J J J J J `aaasG >9>:7F;1A'F$ F;$F( (F;+F( ,F;;F?F?Nr)rrrrrr%r&r'rrs8  4b4b4b4b4b4br&rcdg}|\}}t|trAt|jtr'|j}t|t s |j|vrdSdS)z7A helper function to filter pulses on control channels.rTF)r>r rYr rOr pulse_type time_inst_tup valid_types_instrOs r'_filter_cr_tonerAsh#$KGAt$*T\>"J"J  eX & & %*:k*I*I4 5r&cdg}|\}}t|trAt|jtr'|j}t|t s |j|vrdSdS)z5A helper function to filter pulses on drive channels.rTF)r>r rYrrOrrrs r'_filter_comp_tonerMsh#$KGAt$*T\<"H"H  eX & & %*:k*I*I4 5r&inst_sched_maprr8 Sequence[int]r:(tuple[CRCalType, list[Play], list[Play]]c d}tj5tjdt|d|r|d|}n|d|r$|d|}t j}n|dtt|r=|dtt|}t j }ntd|d|d t|tgjD}d t|tgjD}dddn #1swxYwY|gt!|d krtt#|d |d jj|d kr t j}n t j}t!|dkrt!|dvr|||fSt!|d krMt!|d kr:|d jj|d krt j||fSt j||fStt#|d)anA helper function to check type of CR calibration. Args: inst_sched_map: instruction schedule map of the backends qubits: ordered tuple of qubits for cross resonance (q_control, q_target) Returns: Filtered instructions and most-likely type of calibration. Raises: QiskitError: Unknown calibration type is detected. Nrnror<)r8r=z;Native direction cannot be determined: operation on qubits z- for the following instruction schedule map: cg|] }|d Srr%.0ts r' z+_check_calibration_type..|s```QAaD```r&cg|] }|d Srr%rs r'rz+_check_calibration_type..}sdddqaddddr&rzM has no target compensation tones. Native ECR direction cannot be determined.rrL)rrLz[ is undefined pulse sequence. Check if this is a calibration for cross resonance operation.)rrrrhasrrrtuplereversedr r rrr?rlenreprrYindexr!r"r#r$)rr8rcr_schedrrs r'rrYsH  " "eeX8JKKKK   dF + + %))$v)>>HH   v . . %))%)??H ,HH   uXf-=-='>'> ? ? %))%x7G7G1H1HIIH ,HHQfQQ@NQQ  a`"5h@Q"R"R"_```dd$7CTBU$V$V$cddd )eeeeeeeeeeeeeee, z??a  >>===  a= &&) 3 3 /HH /H 8}}c*oo77:-- 8}}c*oo22 a= &&) 3 3.*D D.*D D  >> H H H  sEE22E69E6)rrr8rr:r)/r __future__renumrRrcollections.abcrrrnumpyrUqiskit.circuitrr7%qiskit.circuit.library.standard_gatesrqiskit.exceptionsr qiskit.pulser r r r rrrrqiskit.pulse.filtersr%qiskit.pulse.instruction_schedule_maprqiskit.transpiler.targetrqiskit.utils.deprecate_pulser base_builderr exceptionsrEnumrr)rrrrr%r&r'rsf """""" $$$$$$555555999999))))))! 444444HHHHHH++++++CCCCCC,,,,,,//////@@@@@ @@@AAAAA.AAAHBbBbBbBbBb"7BbBbBbJ      BBBBBBr&