a h/@sjddlmZddlZddlZddlZddlmZddlmZm Z m Z ddl m Z m Z e rlddlZddlmZejddkredd d d d d Znd d d dd ZGdddeZGdddeZGdddeZGdddeZGdddeZGdddeZGdddeZGdddeZGdd 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_OPS1rNone)targetreturncCsJttdd"}|t|dWdn1s<0YdS)Nz/missing_ops.txta )opentempfile gettempdirwritestr)r fdrA/var/www/auris/lib/python3.9/site-packages/torch/_inductor/exc.py_record_missing_opsrcCsdS)Nr)r rrrrsc@s$eZdZedddddddZdS) OperatorIssuer list[Any]dict[str, Any]rr argskwargsr cCsDd|gddt|D}|r2|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)__name__ __module__ __qualname__ staticmethodr)rrrrrsrcs(eZdZdddddfdd ZZS)MissingOperatorWithoutDecomprrrr rcs(t|td||||dS)Nzmissing lowering )rsuper__init__r)selfr rr __class__rrr0*sz%MissingOperatorWithoutDecomp.__init__r*r+r,r0 __classcell__rrr3rr.)sr.cs(eZdZdddddfdd ZZS)MissingOperatorWithDecomprrrr 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 )rr/r0r)r%dedentr1r3rrr00sz"MissingOperatorWithDecomp.__init__r5rrr3rr7/sr7cs*eZdZddddddfdd ZZS) LoweringException Exceptionrrrr )excr rrr c s0tt|jd|d||||dS)N: r)r/r0typer*r))r2r;r rrr3rrr0@s"zLoweringException.__init__r5rrr3rr9?sr9c@s eZdZdS)SubgraphLoweringExceptionNr*r+r,rrrrr>Hsr>cs"eZdZddfdd ZZS)InvalidCxxCompilerr )r cs.ddlm}td|jd|jjdS)N)configz!No working C++ compiler found in z .cpp.cxx: )rBr/r0r*cppcxx)r2rBr3rrr0Ms zInvalidCxxCompiler.__init__r5rrr3rr@Lsr@cs$eZdZdddfdd ZZS)CppWrapperCodegenErrorrr )msgr cstd|dS)NzC++ wrapper codegen error: r/r0)r2rGr3rrr0VszCppWrapperCodegenError.__init__r5rrr3rrFUsrFcs&eZdZddddfdd ZZS)CppCompileErrorz list[str]rr )cmdoutputr cs>t|tr|d}ttdjd ||ddS)Nzutf-8z C++ compile error Command: {cmd} Output: {output}  )rJrK) isinstancebytesdecoder/r0r%r8stripformatr')r2rJrKr3rrr0[s   zCppCompileError.__init__r5rrr3rrIZsrIc@s eZdZdS)CUDACompileErrorNr?rrrrrRpsrRcs$eZdZdddfdd ZZS) TritonMissingOptional[types.FrameType]r )first_useful_framer 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/tritonrUrH)r2rUr3rrr0uszTritonMissing.__init__r5rrr3rrStsrScs&eZdZddddfdd ZZS)GPUTooOldForTritonr rTr ) device_propsrUr 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 .rV)r/r0namemajorminor)r2rXrUr3rrr0s zGPUTooOldForTriton.__init__r5rrr3rrW~srWc@s"eZdZdZddddddZdS) InductorErrorZinductorr:rTr )inner_exceptionrUr cCs*||_tj|t|jd||ddS)Nr<rV)r^rr0r=r*)r2r^rUrrrr0s zInductorError.__init__N)r*r+r, backend_namer0rrrrr]sr])! __future__rosrr% functoolsrtypingrrrZtorch._dynamo.excrrtypesZ torch.cudar environgetr RuntimeErrorrr.r7r9r>r@rFrIrRrSrWr]rrrrs2