Bj.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 V/var/www/html/Carbon-Document/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 eZdZdfd ZxZS)InvalidCxxCompilerr rcddlm}td|jd|jjdS)N)configz!No working C++ compiler found in z .cpp.cxx: )rWr8r9r1cppcxx)r;rWr<s rr9zInvalidCxxCompiler.__init__SsT  [ [ [6:> [ [     r)r rr>r@s@rrTrTRs=          rrTc eZdZdfd ZxZS)CppWrapperCodegenErrorrNrr rcPtd|dS)NzC++ wrapper codegen error: r8r9)r;rNr<s rr9zCppWrapperCodegenError.__init__\s* 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}  )rarc) isinstancebytesdecoderarcr8r9r,rDstripformatr.)r;rarcr<s rr9zCppCompileError.__init__as fe $ $ ,]]7++F   O  UWW V fV 5 5     r"tuple[type, tuple[list[str], str]]c,|j|j|jffSr)r<rarc)r;s r __reduce__zCppCompileError.__reduce__xs4; 788r)rarbrcrr r)r rk)r1r2r3r9rmr?r@s@rr`r``sQ      .99999999rr`ceZdZdS)CUDACompileErrorNrQrrrroro|rRrroc 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/tritonrrr^)r;rrr<s rr9zTritonMissing.__init__s9  k 2      r)rrrsr rr>r@s@rrqrqs=          rrqc eZdZdfd ZxZS) GPUTooOldForTriton device_propsr rrrsr 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 .ru)r8r9namemajorminor)r;rxrrr<s rr9zGPUTooOldForTriton.__init__sp  _\& _ _5A5G _ _JVJ\ _ _ 2      r)rxr rrrsr rr>r@s@rrwrws=           rrwceZdZdZd dZd S) InductorErrorinductorinner_exceptionrHrrrsr rcp||_tj|t|jd||dS)NrLru)rrr9rMr1)r;rrrs rr9zInductorError.__init__sR /! O$$- B B B B1      rN)rrHrrrsr r)r1r2r3 backend_namer9rrrrrs-L       rr)r rr r) __future__rosrr, functoolsrtypingrrtorch._dynamo.excrrtypes torch.cudar environgetr RuntimeErrorrr6rBrFrPrTr\r`rorqrwrrrrrs"""""" %%%%%%%%EEEEEEEE1LLL000000:>>344;;Yt__)))_))     77777L777YYYYY=YYY                                >>>>>\>>> 99999l9998             $         )          )      r