o Zh/@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)_CudaDevicePropertiesZTORCHINDUCTOR_WRITE_MISSING_OPS1targetrreturnNonecCsLttdd}|t|dWddS1swYdS)Nz/missing_ops.txta )opentempfile gettempdirwritestr)r fdrB/var/www/auris/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 r cs(t|td||||dS)Nzmissing lowering )rsuper__init__r(selfr rr __class__rrr0*s z%MissingOperatorWithoutDecomp.__init__r rrrrrr r r)r*r+r0 __classcell__rrr3rr.)r.cr-) MissingOperatorWithDecompr rrrrrr r cs: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 )rr/r0r(r$dedentr1r3rrr00sz"MissingOperatorWithDecomp.__init__r5r6rrr3rr9/r8r9cseZdZd fd d ZZS)LoweringExceptionexc Exceptionr rrrrrr r c s0tt|jd|d||||dS)N: r)r/r0typer)r()r2r<r rrr3rrr0@s"zLoweringException.__init__) r<r=r rrrrrr r r6rrr3rr;?r8r;c@ eZdZdS)SubgraphLoweringExceptionNr)r*r+rrrrrAHrAcseZdZdfdd ZZS)InvalidCxxCompilerr r cs.ddlm}td|jd|jjdS)N)configz!No working C++ compiler found in z .cpp.cxx: )rFr/r0r)cppZcxx)r2rFr3rrr0Ms zInvalidCxxCompiler.__init__)r r r6rrr3rrDLr8rDceZdZdfdd ZZS)CppWrapperCodegenErrormsgrr r cstd|dS)NzC++ wrapper codegen error: r/r0)r2rKr3rrr0VszCppWrapperCodegenError.__init__)rKrr r r6rrr3rrJUr8rJceZdZd fdd ZZS) CppCompileErrorcmd list[str]outputrr r cs>t|tr |d}ttdjd ||ddS)Nzutf-8z C++ compile error Command: {cmd} Output: {output}  )rOrQ) isinstancebytesdecoder/r0r$r:stripformatr&)r2rOrQr3rrr0[s   zCppCompileError.__init__)rOrPrQrr r r6rrr3rrNZr8rNc@r@)CUDACompileErrorNrBrrrrrXprCrXcrI) TritonMissingfirst_useful_frameOptional[types.FrameType]r r cstjd|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/tritonrZrL)r2rZr3rrr0us zTritonMissing.__init__)rZr[r r r6rrr3rrYtr8rYcrM) GPUTooOldForTriton device_propsr rZr[r r cs,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\)r/r0namemajorminor)r2r^rZr3rrr0s  zGPUTooOldForTriton.__init__)r^r rZr[r r r6rrr3rr]~r8r]c@seZdZdZd dd Zd S) InductorErrorZinductorinner_exceptionr=rZr[r r cCs*||_tj|t|jd||ddS)Nr>r\)rdrr0r?r))r2rdrZrrrr0s  zInductorError.__init__N)rdr=rZr[r r )r)r*r+ backend_namer0rrrrrcsrc)r rr r )! __future__rosrr$ functoolsrtypingrrrZtorch._dynamo.excrrtypesZ torch.cudar environgetr RuntimeErrorrr.r9r;rArDrJrNrXrYr]rcrrrrs4