U ʹj&@sDddlZddlZddlmZmZddlmZmZmZm Z ddl m Z m Z ddl ZddlmZddlmZddlmZerdd lmZnd d ZddlZeeZejed ZGd ddeZGdddeZGdddeZ GdddeZ!GdddeZ"GdddeZ#GdddeZ$GdddeZ%GdddeZ&Gdd d eZ'Gd!d"d"e&Z(Gd#d$d$e&Z)Gd%d&d&e(Z*Gd'd(d(eZ+Gd)d*d*e&Z,Gd+d,d,e-Z.ej/j0j1ej/j0j2ej/j0j3ej/j0j4fZ5d-d.Z6e7d/d0d1Z8e7d/d2d3Z9Gd4d5d5Z:dCd8d9Z;dDe e d:d;d<ZZ=dEd?d@Z>dFdAdBZ?dS)GN)autoEnum) extract_stack format_exc format_list StackSummary)castOptional)config) is_fbcode)counters)exportdb_error_messagecCsdS)N) case_namerrP/var/www/html/bwave/python/venv/lib/python3.8/site-packages/torch/_dynamo/exc.pyrsrZ graph_breaksc@s eZdZdS)TorchDynamoExceptionN__name__ __module__ __qualname__rrrrrsrc@s eZdZdS)InternalTorchDynamoErrorNrrrrrr src@s eZdZdS)RestartAnalysisNrrrrrr$src@s eZdZdS) SkipFrameNrrrrrr(src@s eZdZdS)TorchRuntimeErrorNrrrrrr,srcseZdZfddZZS)InvalidBackendcstd|ddS)NzInvalid backend: z=, see `torch._dynamo.list_backends()` for available backends.super__init__)selfname __class__rrr1s zInvalidBackend.__init__rrrr __classcell__rrr"rr0srcseZdZfddZZS) ResetRequiredcsttddS)Nz Must call `torch._dynamo.reset()` before changing backends. Detected two calls to `torch.compile()` with a different backend compiler arguments. )rrtextwrapdedentr r"rrr8s zResetRequired.__init__r$rrr"rr&7sr&cseZdZfddZZS)BackendCompilerFailedcsBt|dd|_||_d|jdt|jd|}t|dS)Nr?zbackend=z raised: z: )getattr backend_nameinner_exceptiontyperrr)r Z backend_fnr.msgr"rrrDszBackendCompilerFailed.__init__r$rrr"rr*Csr*cs.eZdZfddZddZdddZZS) Unsupportedcs2t|tjj|_||_d|_| dSN) rrtorchZ_guardsZTracingContextr real_stackr0category add_to_statsr r0r"rrrLs  zUnsupported.__init__cCs>t|j|jd8<t|j|jdkr:t|j|j=dS)Nr r)r r5r0r)rrrremove_from_statsSszUnsupported.remove_from_stats unimplementedcCs ||_t||jd7<dS)Nr )r5r r0)r r5rrrr6XszUnsupported.add_to_stats)r9)rrrrr8r6r%rrr"rr1Ks r1c@s eZdZdS)RecompileErrorNrrrrrr:]sr:cseZdZfddZZS)ArgsMismatchErrorcst|dSr2rr7r"rrrbszArgsMismatchError.__init__r$rrr"rr;asr;cseZdZfddZZS)AttributeMutationErrorcst|dSr2rr7r"rrrgszAttributeMutationError.__init__r$rrr"rr<fsr<cs eZdZdZfddZZS)CondOpArgsMismatchErrorz? Internal error from cond() due to arguments mismatch. cst|dSr2rr7r"rrrpsz CondOpArgsMismatchError.__init__)rrr__doc__rr%rrr"rr=ksr=c@s0eZdZeZeZeZeZeZeZ dS) UserErrorTypeN) rrrrZDYNAMIC_CONTROL_FLOWZ ANTI_PATTERNZSTANDARD_LIBRARYZCONSTRAIN_VIOLATIONZ DYNAMIC_DIMZ INVALID_INPUTrrrrr?ts r?cs$eZdZdedfdd ZZS) UserErrorN) error_typecs>|dk r"t|tst|t|7}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) isinstancestrAssertionErrorrrrrAmessage)r rAr0rr"rrr~s   zUserError.__init__)N)rrrr?rr%rrr"rr@}sr@c@s eZdZdS)IncorrectUsageNrrrrrrFsrFcCs.t||}td|t|t||dS)Nz%s)format_error_msg_verbosegraph_breaks_logdebuglogwarningr9)ecoder0Zgraph_break_msgrrrunimplemented_with_warnings   rNr0cCs"|tjddkstt|dS)NBREAKF)osenvirongetrDr1rOrrrr9sr9cCs.td|d7<|tjddks*tdS)Nwarningsr rPF)r rQrRrSrDrOrrrrKsrKc@s*eZdZddZddZedddZdS) KeyErrorMsgcCs ||_dSr2)value)r rVrrrrszKeyErrorMsg.__init__cCs t|jSr2)rCrVr)rrr__str__szKeyErrorMsg.__str__returncCs|Sr2)rWr)rrr__repr__szKeyErrorMsg.__repr__N)rrrrrWrCrZrrrrrUsrU FcCsNddl}t|}|dk r6|dd|t|7}tjr`t|dr`|d|jd|jd7}tjsxt|drx|d 7}t|d rt|j d rt|j d r|d |j j d|j j d7}n|d |j j d7}tj s|s|d7}t |jdkrdn t|jd}t|tr,t||f|jdd|_n||}|f|jdd|_dS)Nrz from user code: rrecord_filenamez! Last frame execution written to zD. To run only this frame while debugging, run torch._dynamo.replay('z'). r4zI Set TORCH_LOGS="+dynamo" and TORCHDYNAMO_VERBOSE=1 for more information r. minifier_path buck_commandz Minifier script written to zW. Run this buck command to find the smallest traced graph which reproduces this error: r[zQ. 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 ) tracebackget_real_stackjoinrr Zreplay_record_enabledhasattrr\verboser.r]r^Zsuppress_errorslenargsrCrBKeyErrorrU)excr0Zexportr_r4Zold_msgZnew_msgrrraugment_exc_messages<      rhrXcCs<t|dd}|dkrdSg}|dk r.tt}tt||S)Nr4)r, filter_stackrrr)rgframer4Zstack_above_dynamorrrr`s  r`cCs@g}|D]2}d|jkrqsX        0