Vh/ddlmZddlZddlZddlZddlmZddlmZm 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 C/home/dcms/DCMS/lib/python3.12/site-packages/torch/_inductor/exc.py_record_missing_oprsJ X((*++;x@  rr9r;r=s@rr?r?/s    rr?c4eZdZ dfd ZxZS)LoweringExceptionc |t|t|jd|d|j |||y)N: r)r4r5typer-r))r7excrr$r%r8s rr5zLoweringException.__init__@sA Cy!!""SED,=,=fdF,S+T U r) rG Exceptionrrr$r*r%r+r,r:r;r=s@rrCrC?s5  &) 1: DR   rrCc eZdZy)SubgraphLoweringExceptionNr-r.r/rrrrJrJHrrJc eZdZdfd ZxZS)InvalidCxxCompilercxddlm}t| d|jd|j j y)N)configz!No working C++ compiler found in z .cpp.cxx: )rQr4r5r-cppcxx)r7rQr8s rr5zInvalidCxxCompiler.__init__Ms1 //@ 6::>>JZ [ r)r,r:r;r=s@rrNrNLs   rrNc eZdZdfd ZxZS)CppWrapperCodegenErrorc*t|d|y)NzC++ wrapper codegen error: r4r5)r7msgr8s rr5zCppWrapperCodegenError.__init__Vs 6se<=r)rYrr,r:r;r=s@rrVrVUs >>rrVc eZdZdfd ZxZS)CppCompileErrorct|tr|jd}t|t j djjdj||y)Nzutf-8z C++ compile error Command: {cmd} Output: {output}  )cmdoutput) isinstancebytesdecoder4r5r!rAstripformatr#)r7r^r_r8s rr5zCppCompileError.__init__[sX fe $]]7+F  OO UW V fV 5 r)r^z list[str]r_rr,r:r;r=s@rr[r[Zs   rr[c eZdZy)CUDACompileErrorNrKrrrrfrfprLrrfc 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_framerX)r7rkr8s rr5zTritonMissing.__init__us   k 2  r)rkOptional[types.FrameType]r,r:r;r=s@rrhrhts   rrhc,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 .rj)r4r5namemajorminor)r7 device_propsrkr8s rr5zGPUTooOldForTriton.__init__sN \&&'(55A5G5G4H,J\J\I] _ 2  r)rtr rkrlr,r:r;r=s@rrnrn~s)  +  6      rrnc$eZdZdZ ddZy) InductorErrorinductorcr||_tj|t|jd||y)NrErj)inner_exceptionr r5rFr-)r7ryrks rr5zInductorError.__init__s; /!! O$--.b0A B1 rN)ryrHrkrlr,r:)r-r.r/ backend_namer5rrrrvrvs(L  "  6     rrv)rrr,r:)! __future__rosrr! functoolsrtypingrrrtorch._dynamo.excrr types torch.cudar environgetr RuntimeErrorrr2r?rCrJrNrVr[rfrhrnrvrrrrs" //E0::>>34;t_))  7L7Y=Y         >\>  l ,   $   )    )  r