iddlmZddlZddlZddlZddlmZddlmZm Z m Z ddl m Z m Z e r ddlZddlmZej"j%ddk(redd#d Znd#d ZGd d eZGd deZGddeZGddeZGddeZGddeZGddeZGddeZGddeZGdde ZGdd e ZGd!d"e Z y)$) annotationsN) lru_cache)AnyOptional TYPE_CHECKING)BackendCompilerFailedShortenTraceback)_CudaDevicePropertiesTORCHINDUCTOR_WRITE_MISSING_OPS1cttjdd5}|jt |dzdddy#1swYyxYw)Nz/missing_ops.txta )opentempfile gettempdirwritestr)targetfds d/var/www/vps2.regionflexible.com/Desarrollo/venv/lib/python3.12/site-packages/torch/_inductor/exc.py_record_missing_oprsJ X((*++;x@  rr:r<r>s@rr@r@/s    rr@c:eZdZ d dfd ZxZS)LoweringExceptionc t|jd|d|j|||}|r ||d|z }t||y)N: rz Found from : )typer.r*r5r6)r8excrr%r& stack_tracemsgr9s rr6zLoweringException.__init__@s\c##$Bse2d.?.?f.U-VW  cU,[M: :C rr) rH Exceptionrrr%r+r&r,rIz Optional[str]r-r;r<r>s@rrDrD?sL&*        #    rrDc eZdZy)SubgraphLoweringExceptionNr.r/r0rrrrMrMNrrMc eZdZdfd ZxZS)InvalidCxxCompilercxddlm}t| d|jd|j j y)N)configz!No working C++ compiler found in z .cpp.cxx: )rTr5r6r.cppcxx)r8rTr9s rr6zInvalidCxxCompiler.__init__Ss1 //@ 6::>>JZ [ r)r-r;r<r>s@rrQrQRs   rrQc eZdZdfd ZxZS)CppWrapperCodegenErrorc*t|d|y)NzC++ wrapper codegen error: r5r6)r8rJr9s rr6zCppWrapperCodegenError.__init__\s 6se<=r)rJrr-r;r<r>s@rrYrY[s >>rrYc(eZdZdfd ZddZxZS)CppCompileErrorct|tr|jd}||_||_t |tjdjjdj||y)Nzutf-8z C++ compile error Command: {cmd} Output: {output}  )cmdoutput) isinstancebytesdecoder`rar5r6r"rBstripformatr$)r8r`rar9s rr6zCppCompileError.__init__asf fe $]]7+F   OO UW V fV 5 rcJ|j|j|jffSr)r9r`ra)r8s r __reduce__zCppCompileError.__reduce__xs4;; 788r)r`z list[str]rarr-r;)r-z"tuple[type, tuple[list[str], str]])r.r/r0r6rhr=r>s@rr]r]`s  .9rr]c eZdZy)CUDACompileErrorNrNrrrrjrj|rOrrjc eZdZdfd ZxZS) TritonMissingc(t|d|y)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/tritonfirst_useful_framer[)r8ror9s rr6zTritonMissing.__init__s   k 2  r)roOptional[types.FrameType]r-r;r<r>s@rrlrls   rrlc,eZdZ dfd ZxZS)GPUTooOldForTritoncvt|d|jd|jd|j|y)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 .rn)r5r6namemajorminor)r8 device_propsror9s rr6zGPUTooOldForTriton.__init__sN \&&'(55A5G5G4H,J\J\I] _ 2  r)rxr rorpr-r;r<r>s@rrrrrs+  ,  6     rrrc$eZdZdZ ddZy) InductorErrorinductorcr||_tj|t|jd||y)NrFrn)inner_exceptionr r6rGr.)r8r}ros rr6zInductorError.__init__s; /!! O$--.b0A B1 rN)r}rKrorpr-r;)r.r/r0 backend_namer6rrrrzrzs(L  "  6     rrz)rrr-r;)! __future__rosrr" functoolsrtypingrrrtorch._dynamo.excrr types torch.cudar environgetr RuntimeErrorrr3r@rDrMrQrYr]rjrlrrrzrrrrs" //E0::>>34;t_))  7L7Y=Y        >\> 9l98   $   )    )  r