P/j8 VddlZddlZddlmZmZddlmZmZmZm Z ddl m Z m Z m Z mZmZmZddlZddlmZddlmZerddlmZd ZddlZej4eZej:j=ed 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,eZ1Gd-d.e,Z2Gd/d0e!Z3Gd1d2e3e,Z4Gd3d4e!Z5Gd5d6e!Z6Gd7d8e7Z8Gd9d:e!Z9Gd;de9Z;Gd?d@e9Ze;e?er>=r$rr>c eZdZy)TorchRuntimeErrorNrr#rrr@r@Ar$rr@c eZdZdfd ZxZS)InvalidBackendc,t|d|dy)NzInvalid backend: z=, see `torch._dynamo.list_backends()` for available backends.r,r-)r.namer0s rr-zInvalidBackend.__init__Fs x'd e rr1r r!r"r-r5r6s@rrBrBEs   rrBc eZdZdfd ZxZS) ResetRequiredcJt|tjdy)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.r0s rr-zResetRequired.__init__Ms!  OO  rr1rFr6s@rrHrHLs   rrHc eZdZdfd ZxZS)BackendCompilerFailedct|dd|_||_d|jdt|jd|}t ||y)Nr ?zbackend=z raised: z: )getattr backend_nameinner_exceptiontyper r,r-)r. backend_fnrRmsgr0s rr-zBackendCompilerFailed.__init__YsW#J C@.**-Z_8M8V8V7WWYZiYjk rr1rFr6s@rrMrMXs rrMc4eZdZdddfdZdZddZxZS) UnsupportedNrct||tjjj |_||_d|_|j||_ yr+) r,r-torch_guardsTracingContextr real_stackrUcategory add_to_statsr)r.rUrr0s rr-zUnsupported.__init__asK --66DDF'+  (1rc|jJt|j|jxxdzcc<t|j|jdkrt|j|j=yy)Nrrr]rrUr.s rremove_from_statszUnsupported.remove_from_statsisc}}((()Q.) DMM "488 , 1'1 2rcN||_t||jxxdz cc<y)Nrr`)r.r]s rr^zUnsupported.add_to_statsos"  488$)$rr1) unimplemented)r r!r"r-rbr^r5r6s@rrWrW`s)-22 *rrWc eZdZy)RecompileErrorNrr#rrrfrftr$rrfc eZdZdfd ZxZS)ArgsMismatchErrorc$t||yr+rDr.rUr0s rr-zArgsMismatchError.__init__y rr1rFr6s@rrhrhx rrhc eZdZdfd ZxZS)AttributeMutationErrorc$t||yr+rDrjs rr-zAttributeMutationError.__init__~rkrr1rFr6s@rrnrn}rlrrnc$eZdZdZdfd ZxZS)CondOpArgsMismatchErrorz? Internal error from cond() due to arguments mismatch. c$t||yr+rDrjs rr-z CondOpArgsMismatchError.__init__rkrr1)r r!r"__doc__r-r5r6s@rrqrqsrrqcneZdZeZeZeZeZeZeZ eZ y) UserErrorTypeN) r r!r"rDYNAMIC_CONTROL_FLOW ANTI_PATTERNSTANDARD_LIBRARYCONSTRAINT_VIOLATION DYNAMIC_DIM INVALID_INPUTINVALID_OUTPUTr#rrrurus666Lv6&KFMVNrruc*eZdZddeddffd ZxZS) UserErrorN error_typer2c|t|dkDr0|d|_|ddj |j |z }t jr,t|dr |d|jd|jdz }t jst|d r|d z }t|d r~t|jd rht|jd r5|d|jjd|jjdz }n|d|jjdz }t js|s|dz }t|jdk(rdnt!|jd}t#|t$r%t'||zf|jddz|_y||z}|f|jddz|_y)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 rR 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_enabledhasattrrverboserRrrsuppress_errorsr/r3rKeyErrorr)rrUrrr\old_msgnew_msgs raugment_exc_messager>s'+C$$J#j/A"5+5b>( %bggi.C.CJ.O&P%QRR ##5F(G 3C4G4G3HI++,E3 3 >>gc<8 \\s%&7 _, 3&& 7 /0C0C0Q0Q/RS0030C0C0P0P/QQSU C /0C0C0Q0Q/RS__ C  ! !&  @ MQ&bC ,>rcg}|D]A}d|jvr|Sd|jvsd|jvr1|j|C|S)N convert_frame eval_frameztorch._dynamo.optimize()rliner)stack user_stackrs rrrs`J! enn ,   5>> )-F%**-T % ! rcd|jd|jd|jd}|dz }|tz }t ||}|,|dz }|dj t |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)rrrrrUr\s rrrs a(8(8'9@S@S?TTVW CCC:<CU+J    rww{:.// t  x Jrc tjdz}tjrt ||||}|Sd|j d|j d|jdt}|S)Nrrrz due to: ) rlineseprrrrrrr)rrrrrUs rformat_error_msgrso **q.C ~~&sD/5I Jt||nAd.>.>-?@ =8 Jr)rFr+)NN)YrrJenumrrrrrrrtypingr r r r r r torch._guardsrYrrutilsrrrlogging getLoggerr r_logginggetArtifactLoggerr RuntimeErrorrr&r(r8r:r<r>r@rBrHrMrWrfrhrnrqrur~rrrr Exceptionrrrrr StopIterationrAttributeErrorrrr _subclasses fake_tensorDataDependentOutputExceptionDynamicOutputShapeExceptionUnsupportedOperatorExceptionUnsupportedFakeTensorException!exceptions_allowed_to_be_fallbackr3robjectrrdrrboolrintrrrrrr#rrr sI JJFF'g!>>33HnM <  3  *    /    $  ,  )   (  0*&*( )   [ /D , !5  3[  2  #7  Y  ,  1  (  . , *$< !!>> !!== !!>> !!@@ %! #) # # # 8"*d0 006>sm0 0111.-Y.-S.-.-RV.-b  )  8C=(3- '( ? ?(<2H?< 7; 0 ) PS r