o I h/@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 fdrG/var/www/vscode/kcb/lib/python3.10/site-packages/torch/_inductor/exc.py_record_missing_ops"rcCsdS)Nr)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__rrr1*s z%MissingOperatorWithoutDecomp.__init__r rrrrrr rr*r+r,r1 __classcell__rrr4rr/)r/cr.) 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 )rr0r1r)r%dedentr2r4rrr10sz"MissingOperatorWithDecomp.__init__r6r7rrr4rr:/r9r:cseZdZd fd d ZZS)LoweringExceptionexc Exceptionr rrrrrr rc s0tt|jd|d||||dS)N: r)r0r1typer*r))r3r=r rrr4rrr1@s"zLoweringException.__init__) r=r>r rrrrrr rr7rrr4rr<?r9r<c@ eZdZdS)SubgraphLoweringExceptionNr*r+r,rrrrrBHrBcseZdZdfdd ZZS)InvalidCxxCompilerr rcs.ddlm}td|jd|jjdS)N)configz!No working C++ compiler found in z .cpp.cxx: )rGr0r1r*cppcxx)r3rGr4rrr1Ms zInvalidCxxCompiler.__init__)r rr7rrr4rrELr9rEceZdZdfdd ZZS)CppWrapperCodegenErrormsgrr rcstd|dS)NzC++ wrapper codegen error: r0r1)r3rMr4rrr1VszCppWrapperCodegenError.__init__)rMrr rr7rrr4rrLUr9rLceZdZd fdd ZZS) CppCompileErrorcmd list[str]outputrr rcs>t|tr |d}ttdjd ||ddS)Nzutf-8z C++ compile error Command: {cmd} Output: {output}  )rQrS) isinstancebytesdecoder0r1r%r;stripformatr')r3rQrSr4rrr1[s   zCppCompileError.__init__)rQrRrSrr rr7rrr4rrPZr9rPc@rA)CUDACompileErrorNrCrrrrrZprDrZcrK) 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\rN)r3r\r4rrr1us zTritonMissing.__init__)r\r]r rr7rrr4rr[tr9r[crO) GPUTooOldForTriton device_propsr r\r]r 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 .r^)r0r1namemajorminor)r3r`r\r4rrr1s  zGPUTooOldForTriton.__init__)r`r r\r]r rr7rrr4rr_~r9r_c@seZdZdZd dd Zd S) InductorErrorinductorinner_exceptionr>r\r]r rcCs*||_tj|t|jd||ddS)Nr?r^)rgrr1r@r*)r3rgr\rrrr1s  zInductorError.__init__N)rgr>r\r]r r)r*r+r, backend_namer1rrrrresre)r rr r)! __future__rosrr% functoolsrtypingrrrtorch._dynamo.excrrtypes torch.cudar environgetr RuntimeErrorrr/r:r<rBrErLrPrZr[r_rerrrrs4