o I&ih* @sddlZddlZddlmZmZddlmZmZmZm Z ddl m Z m Z m Z ddlZddlmZddlmZddlmZerFdd lmZnd d ZddlZeeZejed ZGd ddeZGdddeZ GdddeZ!Gddde!Z"Gddde!Z#GdddeZ$GdddeZ%GdddeZ&GdddeZ'Gdd d eZ(Gd!d"d"eZ)Gd#d$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/d0d0eZ0Gd1d2d2e1Z2ej3j4j5ej3j4j6ej3j4j7ej3j4j8fZ9d3e1d4e:d5e fd6d7Z;d4e:d5e fd8d9ZdLd@e1d4e:dAe?d5dfdBdCZ@dMd@e1d5e e fdDdEZAdFdGZB dNd@e1d5e:fdHdIZCdNd@e1d5e:fdJdKZDdS)ON)autoEnum) extract_stack format_exc format_list StackSummary)castNoReturnOptional)config) is_fbcode)counters)exportdb_error_messagecCsd|ddS)NzhFor more information about this error, see: https://pytorch.org/docs/main/generated/exportdb/index.html#_-)replace) case_namerrr9r2rrrremove_from_stats_s zUnsupported.remove_from_stats unimplementedcCs ||_t||jd7<dS)Nr rA)r(r>rrrr?eszUnsupported.add_to_stats)rC)rrrr'rBr?r-rrr*rr:Ws r:c@r)RecompileErrorNrrrrrrDjrrDcr#)ArgsMismatchErrorct|dSr;r%r@r*rrr'ozArgsMismatchError.__init__r,rrr*rrEnr.rEcr#)AttributeMutationErrorcrFr;r%r@r*rrr'trGzAttributeMutationError.__init__r,rrr*rrHsr.rHcs eZdZdZfddZZS)CondOpArgsMismatchErrorz? Internal error from cond() due to arguments mismatch. crFr;r%r@r*rrr'}rGz CondOpArgsMismatchError.__init__)rrr__doc__r'r-rrr*rrIxsrIc@s0eZdZeZeZeZeZeZeZ dS) UserErrorTypeN) rrrrZDYNAMIC_CONTROL_FLOWZ ANTI_PATTERNZSTANDARD_LIBRARYZCONSTRAINT_VIOLATIONZ DYNAMIC_DIMZ INVALID_INPUTrrrrrKs rKcs$eZdZddeffdd ZZS) UserErrorN error_typecsZ|durt|ts J|dr|d7}n|d7}|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.  ) isinstancestrendswithrr&r'rMmessage)r(rMr9rr*rrr's      zUserError.__init__r;)rrrrKr'r-rrr*rrLsrLc@r)UncapturedHigherOrderOpErrorNrrrrrrUrrUc@r)IncorrectUsageNrrrrrrVrrVer9returncCs*t||}td|t|t||)Nz%s)format_error_msg_verbosegraph_breaks_logdebuglogwarningrC)rWcoder9Zgraph_break_msgrrrunimplemented_with_warnings    r_cCs|tjddks Jt|)NBREAKF)osenvirongetr:r9rrrrCsrCcCs.td|d7<|tjddksJdS)Nwarningsr r`F)rrarbrcrdrrrr]sr]c@s*eZdZddZddZdefddZdS) KeyErrorMsgcCs ||_dSr;)value)r(rgrrrr' zKeyErrorMsg.__init__cCs t|jSr;)rRrgr2rrr__str__rhzKeyErrorMsg.__str__rXcCs|Sr;)rir2rrr__repr__szKeyErrorMsg.__repr__N)rrrr'rirRrjrrrrrfsrfrPFexcexportcCsJddl}t|}|dur|dd||7}tjr.t|dr.|d|jd|jd7}tjs:t|dr:|d 7}t|d ret|j d ret|j d r[|d |j j d|j j d7}n |d |j j d7}tj sn|sn|d7}t |jdkrwdnt|jd}t|trt||f|jdd|_dS||}|f|jdd|_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 r7 minifier_path buck_commandz Minifier script written to zW. Run this buck command to find the smallest traced graph which reproduces this error: rPzQ. 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_enabledhasattrrnverboser7rorpZsuppress_errorslenargsrRrQKeyErrorrf)rkr9rlrqr=Zold_msgZnew_msgrrraugment_exc_messages>        "rycCs<t|dd}|dur dSg}|durtt}tt||S)Nr=)r5 filter_stackrrr)rkframer=Zstack_above_dynamorrrrrs  rrcCsBg}|D]}d|jvr|Sd|jvsd|jvrq||q|S)NZ convert_frameZ eval_frameztorch._dynamo.optimize()filenamelineappend)stackZ user_stackr{rrrrz"s  rzcCspd|jd|jd|jd}|d7}|t7}t||}|dur6|d7}|dt|7}|d7}|d7}|S) NWON'T CONVERT rO line rPz.========== TorchDynamo Stack Trace ========== zY ========== The above exception occurred while processing the following code ========== rmz ==========)co_name co_filenameco_firstlinenorrrrsr)rkr^rnr{r9r=rrrrY.s  rYcCsJtjd}tjrt||||}|Sd|jd|jd|jdt}|S)NrrOrz due to: ) ralinesepr rurYrrrr)rkr^rnr{r9rrrformat_error_msgFs r)rPFr;)NN)Erar0enumrrrqrrrrtypingrr r Z torch._guardsr<rmr r utilsrZtorch.fb.exportdb.loggingrlogging getLoggerrr\Z_loggingZgetArtifactLoggerrZ RuntimeErrorrrrrr r!r"r$r/r3r:rDrErHrIrKrLrU ExceptionrVZ _subclassesZ fake_tensorZDataDependentOutputExceptionZDynamicOutputShapeExceptionZUnsupportedOperatorExceptionZUnsupportedFakeTensorExceptionZ!exceptions_allowed_to_be_fallbackrRr_rCr]rfboolryrrrzrYrrrrrsj        .