o ei@s^ddlmZddlZddlZddlZddlmZddlmZm Z m Z ddl m Z m Z e r6ddlZddlmZejddkrHedd(d dZnd(ddZGdddeZGdddeZGdddeZGdddeZGdddeZGdddeZGdddeZGdddeZGd d!d!eZGd"d#d#e ZGd$d%d%e ZGd&d'd'e Z dS))) annotationsN) lru_cache)AnyOptional TYPE_CHECKING)BackendCompilerFailedShortenTraceback)_CudaDevicePropertiesTORCHINDUCTOR_WRITE_MISSING_OPS1targetrreturnNonecCsLttdd}|t|dWddS1swYdS)Nz/missing_ops.txta )opentempfile gettempdirwritestr)r fdr]/var/www/addictedbytheproject.nl/epg/venv/lib/python3.10/site-packages/torch/_inductor/exc.py_record_missing_ops"rcCsdSNr)r rrrrsc@seZdZed d d Zd S) OperatorIssuer rargs list[Any]kwargsdict[str, Any]r rcCsDd|gddt|D}|r|d|td|dS)Nztarget: cSs g|] \}}d|d|qS)zargs[z]: r).0iargrrr !sz.OperatorIssue.operator_str..zkwargs: rz ) enumerateappendtextwrapindentjoin)r rrlinesrrr operator_strs zOperatorIssue.operator_strN)r rrrrrr r)__name__ __module__ __qualname__ staticmethodr*rrrrrsrceZdZd fd d ZZS) MissingOperatorWithoutDecompr rrrrrr rcs(t|td||||dS)Nzmissing lowering )rsuper__init__r*selfr rr __class__rrr2*s z%MissingOperatorWithoutDecomp.__init__r rrrrrr rr+r,r-r2 __classcell__rrr5rr0)r0cr/) MissingOperatorWithDecompr rrrrrr rcs:t|td||||td|ddS)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 )rr1r2r*r&dedentr3r5rrr20sz"MissingOperatorWithDecomp.__init__r7r8rrr5rr;/r:r;cs"eZdZ ddfdd ZZS)LoweringExceptionNexc Exceptionr rrrrr stack_trace Optional[str]r rc sJt|jd|d||||}|r||d|7}t|dS)N: rz Found from : )typer+r*r1r2)r4r>r rrr@msgr5rrr2@s$zLoweringException.__init__r) r>r?r rrrrrr@rAr rr8rrr5rr=?sr=c@ eZdZdS)SubgraphLoweringExceptionNr+r,r-rrrrrFNrFcseZdZdfdd ZZS)InvalidCxxCompilerr rcs.ddlm}td|jd|jjdS)N)configz!No working C++ compiler found in z .cpp.cxx: )rKr1r2r+cppcxx)r4rKr5rrr2Ss zInvalidCxxCompiler.__init__)r rr8rrr5rrIRr:rIceZdZdfdd ZZS)CppWrapperCodegenErrorrDrr rcstd|dS)NzC++ wrapper codegen error: r1r2)r4rDr5rrr2\szCppWrapperCodegenError.__init__)rDrr rr8rrr5rrP[r:rPcs(eZdZd fdd Zd d d ZZS)CppCompileErrorcmd list[str]outputrr rcsJt|tr |d}||_||_ttd j d ||ddS)Nzutf-8z C++ compile error Command: {cmd} Output: {output}  )rSrU) isinstancebytesdecoderSrUr1r2r&r<stripformatr()r4rSrUr5rrr2as   zCppCompileError.__init__"tuple[type, tuple[list[str], str]]cCs|j|j|jffSr)r6rSrU)r4rrr __reduce__xszCppCompileError.__reduce__)rSrTrUrr r)r r\)r+r,r-r2r]r9rrr5rrR`srRc@rE)CUDACompileErrorNrGrrrrr^|rHr^crO) TritonMissingfirst_useful_frameOptional[types.FrameType]r rcstjd|ddS)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/tritonr`rQ)r4r`r5rrr2s zTritonMissing.__init__)r`rar rr8rrr5rr_r:r_cseZdZd fdd ZZS) GPUTooOldForTriton device_propsr r`rar rcs,tjd|jd|jd|j|ddS)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 .rb)r1r2namemajorminor)r4rdr`r5rrr2s  zGPUTooOldForTriton.__init__)rdr r`rar rr8rrr5rrcr:rcc@seZdZdZd dd Zd S) InductorErrorinductorinner_exceptionr?r`rar rcCs*||_tj|t|jd||ddS)NrBrb)rkrr2rCr+)r4rkr`rrrr2s  zInductorError.__init__N)rkr?r`rar r)r+r,r- backend_namer2rrrrrisri)r rr r)! __future__rosrr& functoolsrtypingrrrtorch._dynamo.excrrtypes torch.cudar environgetr RuntimeErrorrr0r;r=rFrIrPrRr^r_rcrirrrrs4