tj>.ddlmZddlZddlZddlZddlmZddlmZm Z ddl m Z m Z e r ddl Z ddlmZejddkredd'd Znd'dZGddeZGddeZGddeZGddeZGddeZGddeZGddeZGddeZGdd eZGd!d"e ZGd#d$e ZGd%d&e ZdS)() annotationsN) lru_cache)Any TYPE_CHECKING)BackendCompilerFailedShortenTraceback)_CudaDevicePropertiesTORCHINDUCTOR_WRITE_MISSING_OPS1targetrreturnNonecttjdd5}|t |dzddddS#1swxYwYdS)Nz/missing_ops.txta )opentempfile gettempdirwritestr)r fds Q/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/torch/_inductor/exc.py_record_missing_oprs X(**<<z.OperatorIssue.operator_str..!s:) ) ) $*As A  #  ) ) ) rzkwargs: rz ) enumerateappendtextwrapindentjoin)r r r"liness r operator_strzOperatorIssue.operator_strs|$F$$%) ) .7oo) ) )    . LL,F,, - - -tyy//666rN)r rr r!r"r#r r)__name__ __module__ __qualname__ staticmethodr0rrrrrs2777\777rrc eZdZd fd ZxZS) MissingOperatorWithoutDecompr rr r!r"r#r rct|td||||dS)Nzmissing lowering )rsuper__init__r0selfr r r" __class__s rr9z%MissingOperatorWithoutDecomp.__init__*sJ6""" Wd.?.?f.U.UWWXXXXXrr rr r!r"r#r rr1r2r3r9 __classcell__r<s@rr6r6)sGYYYYYYYYYYrr6c eZdZd fd ZxZS) MissingOperatorWithDecompr rr r!r"r#r rct|td||||t jd|dzdS)Nzmissing decomposition z9 There is a decomposition available for z in torch._decomp.get_decompositions(). Please add this operator to the `decompositions` list in torch._inductor.decomposition )rr8r9r0r,dedentr:s rr9z"MissingOperatorWithDecomp.__init__0s6"""  Od&7&7f&M&M O Oo8>  rr=r>r@s@rrBrB/s=           rrBc$eZdZ ddfd ZxZS)LoweringExceptionNexc Exceptionr rr r!r"r# stack_trace str | Noner rc t|jd|d||||}|r ||d|z }t|dS)N: rz Found from : )typer1r0r8r9)r;rGr r r"rImsgr<s rr9zLoweringException.__init__@syc#WWsWWd.?.?f.U.UWW  ; c::[:: :C rr) rGrHr rr r!r"r#rIrJr rr>r@s@rrFrF?sG#'           rrFceZdZdS)SubgraphLoweringExceptionNr1r2r3rrrrPrPNDrrPc"eZdZddfd ZxZS) InvalidCxxCompilerNcompilerrJr rcddlm}|d|jd|jj}nd|d}t |dS)N)configz!No working C++ compiler found in z .cpp.cxx: z Compiler: z is not found.)rXr1cppcxxr8r9)r;rUrXrNr<s rr9zInvalidCxxCompiler.__init__Ssv  $FO$$:>$$ C 8x777C rr)rUrJr rr>r@s@rrTrTRsB           rrTc eZdZdfd ZxZS)CppWrapperCodegenErrorrNrr rcPtd|dS)NzC++ wrapper codegen error: r8r9)r;rNr<s rr9zCppWrapperCodegenError.__init__as* r@s@rr]r]`s=>>>>>>>>>>rr]c(eZdZd fd Zd d ZxZS) CppCompileErrorcmd list[str]outputrr rcRt|tr|d}||_||_t tjd d ||dS)Nzutf-8z C++ compile error Command: {cmd} Output: {output}  )rbrd) isinstancebytesdecoderbrdr8r9r,rDstripformatr.)r;rbrdr<s rr9zCppCompileError.__init__fs fe $ $ ,]]7++F   O  UWW V fV 5 5     r"tuple[type, tuple[list[str], str]]c,|j|j|jffSr)r<rbrd)r;s r __reduce__zCppCompileError.__reduce__}s4; 788r)rbrcrdrr r)r rl)r1r2r3r9rnr?r@s@rraraesQ      .99999999rraceZdZdS)CUDACompileErrorNrQrrrrprprRrrpc eZdZdfd ZxZS) TritonMissingfirst_useful_frametypes.FrameType | Noner rcNtd|dS)NzCannot find a working triton installation. Either the package is not installed or it is too old. More information on installing Triton can be found at: https://github.com/triton-lang/tritonrsr_)r;rsr<s rr9zTritonMissing.__init__s9  k 2      r)rsrtr rr>r@s@rrrrrs=          rrrc eZdZdfd ZxZS) GPUTooOldForTriton device_propsr rsrtr rc~td|jd|jd|j|dS)NzFound z which is too old to be supported by the triton GPU compiler, which is used as the backend. Triton only supports devices of CUDA Capability >= 7.0, but your device is of CUDA capability .rv)r8r9namemajorminor)r;ryrsr<s rr9zGPUTooOldForTriton.__init__sp  _\& _ _5A5G _ _JVJ\ _ _ 2      r)ryr rsrtr rr>r@s@rrxrxs=           rrxceZdZdZd dZd S) InductorErrorinductorinner_exceptionrHrsrtr rcp||_tj|t|jd||dS)NrLrv)rrr9rMr1)r;rrss rr9zInductorError.__init__sR /! O$$- B B B B1      rN)rrHrsrtr r)r1r2r3 backend_namer9rrrrrs-L       rr)r rr r) __future__rosrr, functoolsrtypingrrtorch._dynamo.excrrtypes torch.cudar environgetr RuntimeErrorrr6rBrFrPrTr]rarprrrxrrrrrs"""""" %%%%%%%%EEEEEEEE1LLL000000:>>344;;Yt__)))_))     77777L777YYYYY=YYY                                >>>>>\>>> 99999l9998             $         )          )      r