%Vji2ddlmZddlZddlZddlZddlmZddlmZm Z m Z ddl m Z m Z e r ddlZddlmZejddkredd'd Znd'dZGddeZGddeZGddeZGddeZGddeZGddeZGddeZGddeZGdd eZGd!d"e ZGd#d$e ZGd%d&e Z dS)() annotationsN) lru_cache)AnyOptional TYPE_CHECKING)BackendCompilerFailedShortenTraceback)_CudaDevicePropertiesTORCHINDUCTOR_WRITE_MISSING_OPS1targetrreturnNonecttjdd5}|t |dzddddS#1swxYwYdS)Nz/missing_ops.txta )opentempfile gettempdirwritestr)r fds M/root/voice-cloning/.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$rr)__name__ __module__ __qualname__ staticmethodr1rrrr r s2777\777rr c eZdZd fd ZxZS) MissingOperatorWithoutDecompr rr!r"r#r$rrct|td||||dS)Nzmissing lowering )rsuper__init__r1selfr r!r# __class__s rr:z%MissingOperatorWithoutDecomp.__init__*sJ6""" Wd.?.?f.U.UWWXXXXXrr rr!r"r#r$rrr2r3r4r: __classcell__r=s@rr7r7)sGYYYYYYYYYYrr7c eZdZd fd ZxZS) MissingOperatorWithDecompr rr!r"r#r$rrct|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 )rr9r:r1r-dedentr;s rr:z"MissingOperatorWithDecomp.__init__0s6"""  Od&7&7f&M&M O Oo8>  rr>r?rAs@rrCrC/s=           rrCc eZdZd fd ZxZS) LoweringExceptionexc Exceptionr rr!r"r#r$rrc tt|jd|d||||dS)N: r)r9r:typer2r1)r<rHr r!r#r=s rr:zLoweringException.__init__@s_ Cyy! U US U UD,=,=fdF,S,S U U     r) rHrIr rr!r"r#r$rrr?rAs@rrGrG?=          rrGceZdZdS)SubgraphLoweringExceptionNr2r3r4rrrrOrOHDrrOc eZdZdfd ZxZS)InvalidCxxCompilerrrcddlm}td|jd|jjdS)N)configz!No working C++ compiler found in z .cpp.cxx: )rVr9r:r2cppcxx)r<rVr=s rr:zInvalidCxxCompiler.__init__MsT  [ [ [6:> [ [     r)rrr?rAs@rrSrSLrMrrSc eZdZdfd ZxZS)CppWrapperCodegenErrormsgrrrcPtd|dS)NzC++ wrapper codegen error: r9r:)r<r\r=s rr:zCppWrapperCodegenError.__init__Vs* >>>>>>>>>rr[c(eZdZd fd Zd d ZxZS) CppCompileErrorcmd list[str]outputrrrcRt|tr|d}||_||_t tjd d ||dS)Nzutf-8z C++ compile error Command: {cmd} Output: {output}  )rarc) isinstancebytesdecoderarcr9r:r-rEstripformatr/)r<rarcr=s rr:zCppCompileError.__init__[s 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__rs4; 788r)rarbrcrrr)rrk)r2r3r4r:rmr@rAs@rr`r`ZsQ      .99999999rr`ceZdZdS)CUDACompileErrorNrPrrrrorovrQrroc eZdZdfd ZxZS) TritonMissingfirst_useful_frameOptional[types.FrameType]rrcNtd|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 rr:zTritonMissing.__init__{s9  k 2      r)rrrsrrr?rAs@rrqrqzs=          rrqc eZdZdfd ZxZS) GPUTooOldForTriton device_propsr rrrsrrc~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)r9r:namemajorminor)r<rxrrr=s rr:zGPUTooOldForTriton.__init__sp  _\& _ _5A5G _ _JVJ\ _ _ 2      r)rxr rrrsrrr?rAs@rrwrws=           rrwceZdZdZd dZd S) InductorErrorinductorinner_exceptionrIrrrsrrcp||_tj|t|jd||dS)NrKru)rr r:rLr2)r<rrrs rr:zInductorError.__init__sR /! O$$- B B B B1      rN)rrIrrrsrr)r2r3r4 backend_namer:rrrrrs-L       rr)r rrr)! __future__rosrr- functoolsrtypingrrrtorch._dynamo.excrr types torch.cudar environgetr RuntimeErrorr r7rCrGrOrSr[r`rorqrwrrrrrs"""""" //////////EEEEEEEE1LLL000000:>>344;;Yt__)))_))     77777L777YYYYY=YYY                                 >>>>>\>>> 99999l9998             $         )          )      r