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 M/var/www/html/engine/venv/lib/python3.12/site-packages/torch/_inductor/exc.py_record_missing_oprsJ X((*++;x@  rr:r<r>s@rr@r@/s    rr@c4eZdZ dfd ZxZS)LoweringExceptionc |t|t|jd|d|j |||y)N: r)r5r6typer.r*)r8excrr%r&r9s rr6zLoweringException.__init__@sA Cy!!""SED,=,=fdF,S+T U r) rH Exceptionrrr%r+r&r,r-r;r<r>s@rrDrD?s5  &) 1: DR   rrDc eZdZy)SubgraphLoweringExceptionNr.r/r0rrrrKrKHrrKc eZdZdfd ZxZS)InvalidCxxCompilercxddlm}t| d|jd|j j y)N)configz!No working C++ compiler found in z .cpp.cxx: )rRr5r6r.cppcxx)r8rRr9s rr6zInvalidCxxCompiler.__init__Ms1 //@ 6::>>JZ [ r)r-r;r<r>s@rrOrOLs   rrOc eZdZdfd ZxZS)CppWrapperCodegenErrorc*t|d|y)NzC++ wrapper codegen error: r5r6)r8msgr9s rr6zCppWrapperCodegenError.__init__Vs 6se<=r)rZrr-r;r<r>s@rrWrWUs >>rrWc(eZdZdfd ZddZxZS)CppCompileErrorct|tr|jd}||_||_t |tjdjjdj||y)Nzutf-8z C++ compile error Command: {cmd} Output: {output}  )cmdoutput) isinstancebytesdecoder_r`r5r6r"rBstripformatr$)r8r_r`r9s rr6zCppCompileError.__init__[sf fe $]]7+F   OO UW V fV 5 rcJ|j|j|jffSr)r9r_r`)r8s r __reduce__zCppCompileError.__reduce__rs4;; 788r)r_z list[str]r`rr-r;)r-z"tuple[type, tuple[list[str], str]])r.r/r0r6rgr=r>s@rr\r\Zs  .9rr\c eZdZy)CUDACompileErrorNrLrrrririvrMrric 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_framerY)r8rnr9s rr6zTritonMissing.__init__{s   k 2  r)rnOptional[types.FrameType]r-r;r<r>s@rrkrkzs   rrkc,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 .rm)r5r6namemajorminor)r8 device_propsrnr9s rr6zGPUTooOldForTriton.__init__sN \&&'(55A5G5G4H,J\J\I] _ 2  r)rwr rnror-r;r<r>s@rrqrqs)  +  6      rrqc$eZdZdZ ddZy) InductorErrorinductorcr||_tj|t|jd||y)NrFrm)inner_exceptionr r6rGr.)r8r|rns rr6zInductorError.__init__s; /!! O$--.b0A B1 rN)r|rIrnror-r;)r.r/r0 backend_namer6rrrryrys(L  "  6     rry)rrr-r;)! __future__rosrr" functoolsrtypingrrrtorch._dynamo.excrr types torch.cudar environgetr RuntimeErrorrr3r@rDrKrOrWr\rirkrqryrrrrs" //E0::>>34;t_))  7L7Y=Y         >\> 9l98   $   )    )  r