LjddlmZddlZddlZddlZddlmZddlmZm Z ddl m Z m Z e r ddl Z ddlmZej j#ddk(redd#d Znd#d ZGd d eZGd deZGddeZGddeZGddeZGddeZGddeZGddeZGddeZGdde ZGdd e ZGd!d"e Zy)$) annotationsN) lru_cache)Any TYPE_CHECKING)BackendCompilerFailedShortenTraceback)_CudaDevicePropertiesTORCHINDUCTOR_WRITE_MISSING_OPS1cttjdd5}|jt |dzdddy#1swYyxYw)Nz/missing_ops.txta )opentempfile gettempdirwritestr)targetfds R/var/www/html/ia-enhanced/venv/lib/python3.12/site-packages/torch/_inductor/exc.py_record_missing_oprsJ X((*++;x@  rr9r;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)r4r5)r7excrr$r% stack_tracemsgr8s rr5zLoweringException.__init__@s\c##$Bse2d.?.?f.U-VW  cU,[M: :C rr) rG Exceptionrrr$r*r%r+rHz str | Noner,r:r;r=s@rrCrC?sL#'            rrCc eZdZy)SubgraphLoweringExceptionNr-r.r/rrrrLrLNrrLc eZdZdfd ZxZS)InvalidCxxCompilercxddlm}t| d|jd|j j y)N)configz!No working C++ compiler found in z .cpp.cxx: )rSr4r5r-cppcxx)r7rSr8s rr5zInvalidCxxCompiler.__init__Ss1 //@ 6::>>JZ [ r)r,r:r;r=s@rrPrPRs   rrPc eZdZdfd ZxZS)CppWrapperCodegenErrorc*t|d|y)NzC++ wrapper codegen error: r4r5)r7rIr8s rr5zCppWrapperCodegenError.__init__\s 6se<=r)rIrr,r:r;r=s@rrXrX[s >>rrXc(eZdZdfd ZddZxZS)CppCompileErrorct|tr|jd}||_||_t |tjdjjdj||y)Nzutf-8z C++ compile error Command: {cmd} Output: {output}  )cmdoutput) isinstancebytesdecoder_r`r4r5r!rAstripformatr#)r7r_r`r8s rr5zCppCompileError.__init__asf fe $]]7+F   OO UW V fV 5 rcJ|j|j|jffSr)r8r_r`)r7s r __reduce__zCppCompileError.__reduce__xs4;; 788r)r_z list[str]r`rr,r:)r,z"tuple[type, tuple[list[str], str]])r-r.r/r5rgr<r=s@rr\r\`s  .9rr\c eZdZy)CUDACompileErrorNrMrrrriri|rNrric 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_framerZ)r7rnr8s rr5zTritonMissing.__init__s   k 2  r)rntypes.FrameType | Noner,r:r;r=s@rrkrks   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)r4r5namemajorminor)r7 device_propsrnr8s rr5zGPUTooOldForTriton.__init__sN \&&'(55A5G5G4H,J\J\I] _ 2  r)rwr rnror,r:r;r=s@rrqrqs+  ,  3     rrqc$eZdZdZ ddZy) InductorErrorinductorcr||_tj|t|jd||y)NrErm)inner_exceptionrr5rFr-)r7r|rns rr5zInductorError.__init__s; /!! O$--.b0A B1 rN)r|rJrnror,r:)r-r.r/ backend_namer5rrrryrys(L  "  3     rry)rrr,r:) __future__rosrr! functoolsrtypingrrtorch._dynamo.excrrtypes torch.cudar environgetr RuntimeErrorrr2r?rCrLrPrXr\rirkrqryrrrrs" %E0::>>34;t_))  7L7Y=Y        >\> 9l98   $   )    )  r