j< ddlZddlZddlmZmZddlmZmZmZm Z ddl m Z m Z m Z mZmZmZddlZddlmZddlmZerddlmZd ZddlZejeZejed ZGd d e Z!Gd de!Z"Gdde!Z#Gdde#Z$Gdde#Z%Gdde#Z&Gdde#Z'Gdde!Z(Gdde!Z)Gdde!Z*Gdd e!Z+Gd!d"e!Z,Gd#d$e!Z-Gd%d&e!Z.Gd'd(e-Z/Gd)d*e-Z0Gd+d,e/Z1Gd-d.eZ2Gd/d0e-Z3Gd1d2e!Z4Gd3d4e-Z5Gd5d6e!Z6Gd7d8e!Z7Gd9d:e8Z9Gd;de!Z;Gd?d@e;Z<GdAdBe;Z=GdCdDe;Z>e?eiZBdEZCdFZDejEjFjGejEjFjHejEjFjIejEjFjJfZKdGe8dHeLdIe fdJZMeNZOeOddKdHeLdLe dMeeLdIe fdNZPdHeLdIdfdOZQGdPdQZRd^dTe8dHeLdUeSdIdfdVZTdGe8dWdXdIeeeLeeUffdYZVd_dTe8dIee fdZZWd[ZX d`dTe8dIeLfd\ZYd`dTe8dIeLfd]ZZdS)aN)autoEnum) extract_stack format_exc format_list StackSummary)AnycastNoReturnOptionalTuple TYPE_CHECKING)config)counters) CompileIdc4d|ddzS)NzhFor more information about this error, see: https://pytorch.org/docs/main/generated/exportdb/index.html#_-)replace case_names BC:\Projects\Historical\venv\Lib\site-packages\torch/_dynamo/exc.pyexportdb_error_messagers% I   C % % & graph_breaksceZdZdS)TorchDynamoExceptionN__name__ __module__ __qualname__rrrr!DrrceZdZdS)InternalTorchDynamoErrorNrr#rrr&r&%r$rr&c2eZdZUeed<dddfdZxZS)RestartAnalysisrestart_reasonN)r)returncB||_tj|dSN)r)super__init__)selfr)args __class__s rr.zRestartAnalysis.__init__,s%,$rr*N)r r!r"str__annotations__r. __classcell__r1s@rr(r()sX-1            rr(ceZdZdS)SpeculationRestartAnalysisNrr#rrr8r81r$rr8ceZdZdS)UnspecializeRestartAnalysisNrr#rrr:r:5r$rr:ceZdZdS) CompileCollectiveRestartAnalysisNrr#rrr<r<9r$rr<ceZdZdS)TensorifyScalarRestartAnalysisNrr#rrr>r>=r$rr>ceZdZdS) SkipFrameNrr#rrr@r@Ar$rr@ceZdZdS)TorchRuntimeErrorNrr#rrrBrBEr$rrBc eZdZdfd ZxZS)InvalidBackendr*NcRtd|ddS)NzInvalid backend: z=, see `torch._dynamo.list_backends()` for available backends.r-r.)r/namer1s rr.zInvalidBackend.__init__Js7  e e e e     rr2r r!r"r.r5r6s@rrDrDIs=          rrDc eZdZdfd ZxZS) ResetRequiredr*NcnttjddS)Nz Must call `torch._dynamo.reset()` before changing backends. Detected two calls to `torch.compile()` with a different backend compiler arguments. )r-r.textwrapdedent)r/r1s rr.zResetRequired.__init__Qs=  O       rr2rHr6s@rrJrJPs=          rrJc eZdZdfd ZxZS)BackendCompilerFailedr*Nct|dd|_||_d|jdt|jd|}t |dS)Nr ?zbackend=z raised: z: )getattr backend_nameinner_exceptiontyper r-r.)r/ backend_fnrTmsgr1s rr.zBackendCompilerFailed.__init__]si#J C@@.k*kk_8M8M8VkkZikk rr2rHr6s@rrOrO\s=rrOc4eZdZdddfdZdZd dZxZS) UnsupportedNrr*ct|tjj|_||_d|_| ||_ dSr,) r-r.torch_guardsTracingContextr real_stackrWcategory add_to_statsr)r/rWrr1s rr.zUnsupported.__init__es^ -6DDFF'+  (1rc|jJt|j|jxxdzcc<t|j|jdkrt|j|j=dSdS)Nrrr_rrWr/s rremove_from_statszUnsupported.remove_from_statsmsm}((()))Q.))) DM "48 , 1 1'111 2 1r unimplementedcT||_t||jxxdz cc<dS)Nrrb)r/r_s rr`zUnsupported.add_to_statsss2  48$$$)$$$$$rr2)re)r r!r"r.rdr`r5r6s@rrYrYdsm)-22222222222 ********rrYceZdZdS)RecompileErrorNrr#rrrhrhxr$rrhc eZdZdfd ZxZS)ArgsMismatchErrorr*NcJt|dSr,rFr/rWr1s rr.zArgsMismatchError.__init__}! rr2rHr6s@rrjrj|=rrjc eZdZdfd ZxZS)AttributeMutationErrorr*NcJt|dSr,rFrls rr.zAttributeMutationError.__init__rmrr2rHr6s@rrprprnrrpc$eZdZdZdfd ZxZS)CondOpArgsMismatchErrorz? Internal error from cond() due to arguments mismatch. r*NcJt|dSr,rFrls rr.z CondOpArgsMismatchError.__init__rmrr2)r r!r"__doc__r.r5r6s@rrsrssGrrsceZdZeZeZeZeZeZeZ eZ eZ dS) UserErrorTypeN) r r!r"rDYNAMIC_CONTROL_FLOW ANTI_PATTERNSTANDARD_LIBRARYCONSTRAINT_VIOLATION DYNAMIC_DIM INVALID_INPUTINVALID_OUTPUT*UNSUPPORTED_ALIASED_MUTATED_DYNAMIC_INPUTSr#rrrwrwsl466466Ltvv466$&&KDFFMTVVN15...rrwc*eZdZddeddffd ZxZS) UserErrorN error_typer*c|It|tsJ|dr|dz }n|dz }|t|z }t |||_||_dS)aG Type of errors that would be valid in Eager, but not supported in TorchDynamo. The error message should tell user about next actions. error_type: Type of user error msg: Actionable error message case_name: (Optional) Unique name (snake case) for the usage example in exportdb. N.  ) isinstancer3endswithrr-r.rmessage)r/rrWrr1s rr.zUserError.__init__s  i-- - --||C   s t  ))44 4C $ rr,)r r!r"rwr.r5r6s@rrrsP=$rrceZdZdS)SkipCodeRecursiveExceptionNrr#rrrrr$rrceZdZdS)RecompileLimitExceededNrr#rrrrr$rrceZdZdS)UnsafeScriptObjectErrorNrr#rrrrr$rrceZdZdS)UncapturedHigherOrderOpErrorNrr#rrrrr$rrceZdZdS)IncorrectUsageNrr#rrrrr$rrceZdZdS)FailOnRecompileLimitHitNrr#rrrrr$rrceZdZdS)ObservedExceptionNrr#rrrrDrrc8eZdZUeeed<dfd ZxZS)ObservedUserStopIterationvaluer*Nctdt|dkr|d|_dSd|_dS)Nzunhandled `raise StopIteration`r)r-r.lenr)r/r0kwargsr1s rr.z"ObservedUserStopIteration.__init__sE :;;; t99q==aDJJJDJJJrr2)r r!r"r r r4r.r5r6s@rrrsO C=rrceZdZdS)ObservedKeyErrorNrr#rrrrrrrceZdZdS)ObservedAttributeErrorNrr#rrrrrrrcddlm}|||gi}|j|t |)Nr)BuiltinVariable) variablesr call_function exn_vt_stackappendobserved_exception_map)etxr exception_vts rraise_observed_exceptionrsX******#?1%%33BB??LO<(((  ##rc8|jdSr,)rpop)rs rhandle_observed_exceptionrs2OrrrWr*ct||tjddfdtdt |t||dS)Nartifactc dddS)Ndynamo_graph_break_reasonstring)rGencodingr#r#rrz,unimplemented_with_warning..)s/   rcSr,r#)graph_break_msgsrrz,unimplemented_with_warning..-s?r) metadata_fn payload_fnz%s)from_exc) format_error_msg_verboser[_loggingtrace_structuredgraph_breaks_logdebuglogwarningre)rcoderWrs @runimplemented_with_warningrs/q$77O N##  +*** $4111KK#""""""r)rrrrc|tjddksJ|turt |||t ||)NBREAKFr)osenvironget_NOTHINGrY)rWrrs rrere7sX "*..%00 0 0 0 0x#333A cY / / //rctd|xxdz cc<|tjddksJdS)NwarningsrrF)rrrr)rWs rrr@sJ Z" "*..%00 0 0 0 0 0 0rc.eZdZddZdefdZdefdZdS) KeyErrorMsgr*Nc||_dSr,)r)r/rs rr.zKeyErrorMsg.__init__Hs  rc*t|jSr,)r3rrcs r__str__zKeyErrorMsg.__str__Ks4:rc*|Sr,)rrcs r__repr__zKeyErrorMsg.__repr__Ns||~~rr2)r r!r"r.r3rrr#rrrrGs^#rrrFexcexportcddl}d|_t|}|Nt|dkr;|d|_|dd||z }t jr&t|dr|d|j d|j dz }t j st|d r|d z }t|d r^t|j d rIt|j d r!|d|j j d|j j dz }n|d|j j dz }t js|s|dz }t|jdkrdnt!|jd}t#|t$r*t'||zf|jddz|_dS||z}|f|jddz|_dS)Nrz from user code: record_filenamez! Last frame execution written to zD. To run only this frame while debugging, run torch._dynamo.replay('z'). r^zI Set TORCH_LOGS="+dynamo" and TORCHDYNAMO_VERBOSE=1 for more information rT minifier_path buck_commandz Minifier script written to zW. Run this buck command to find the smallest traced graph which reproduces this error: rzQ. Run this script to find the smallest traced graph which reproduces this error. z You can suppress this exception and fall back to eager by setting: import torch._dynamo torch._dynamo.config.suppress_errors = True r) tracebackinnermost_user_frame_summaryget_real_stackrjoinrrreplay_record_enabledhasattrrverboserTrrsuppress_errorsr0r3rKeyErrorr)rrWrrr^old_msgnew_msgs raugment_exc_messagerRsK'+C$$$J#j//A"5"5+5b>( Rbggi.C.CJ.O.O&P&PRRR #35F(G(G3 3C4G33+333 3 >]gc<88] \\s%&& 7 _,,  3& 7 7  U0C0QUU030C0PUUU CC _0C0Q___ C  ! &   @ MMQ&&bbC ,<, >>rczg}|D]5}d|jvrn)d|jvs d|jvr ||6|S)N convert_frame eval_frameztorch._dynamo.optimize()rliner)stack user_stackrs rrrsbJ!! en , , E 5> ) )-F%*-T-T %    rcd|jd|jd|jd}|dz }|tz }t ||}|4|dz }|dt |z }|dz }|dz }|S) NWON'T CONVERT r line rz.========== TorchDynamo Stack Trace ========== zY ========== The above exception occurred while processing the following code ========== rz ==========)co_name co_filenameco_firstlinenorrrr)rrrrrWr^s rrrs XWW(8WW@SWWW CCC:<<CU++J    rww{:../// t  x Jrc tjdz}tjrt ||||}n)d|jd|jd|jdt}|S)Nrrrz due to: ) rlineseprrrrrrr)rrrrrWs rformat_error_msgrsx *q.C ~8&sD/5II8t|88d.>88 88)388 Jr)rFr,)NN)[rrLenumrrrrrrrtypingr r r r r r torch._guardsr[rrutilsrrrlogging getLoggerr rrgetArtifactLoggerr RuntimeErrorrr&r(r8r:r<r>r@rBrDrJrOrYrhrjrprsrwrrrrr Exceptionrrrrrr StopIterationrAttributeErrorrrr _subclasses fake_tensorDataDependentOutputExceptionDynamicOutputShapeExceptionUnsupportedOperatorExceptionUnsupportedFakeTensorException!exceptions_allowed_to_be_fallbackr3robjectrrerrboolrintrrrrrr#rrrs5 JJJJJJJJJJJJFFFFFFFFFFFFFFFF(''''''g!!>33HnMM     <        3        *                /                _        $        ,        )         (    0*****&***(     )     [ /88888D888 ,     !5        [        2        #7        Y        i        ,        1        (        .   , *$$$< !> != !> !@ %!#)######, 688"*d000 006>sm0 0000111111.-.-Y.-S.-.-RV.-.-.-.-b  )  8C=(3- '(    ?? ?(<2H????<   7; 0  ) PS      r