o j9:j@sZddlmZddlZddlZddlZddlmZddlmZm Z ddl m Z m Z e r4ddl Z ddlmZejddkrFedd(d dZnd(ddZGdddeZGdddeZGdddeZGdddeZGdddeZGdddeZGdddeZGdddeZGd d!d!eZGd"d#d#e ZGd$d%d%e ZGd&d'd'e ZdS))) annotationsN) lru_cache)Any TYPE_CHECKING)BackendCompilerFailedShortenTraceback)_CudaDevicePropertiesTORCHINDUCTOR_WRITE_MISSING_OPS1targetrreturnNonecCsLttdd}|t|dWddS1swYdS)Nz/missing_ops.txta )opentempfile gettempdirwritestr)r fdrZ/home/nk/hobo-godmode/plappi-mvp/.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 r cs(t|td||||dS)Nzmissing lowering )rsuper__init__r)selfr rr __class__rrr1*s z%MissingOperatorWithoutDecomp.__init__r rrrrrr r r*r+r,r1 __classcell__rrr4rr/)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 )rr0r1r)r%dedentr2r4rrr10sz"MissingOperatorWithDecomp.__init__r6r7rrr4rr:/r9r:cs"eZdZ ddfdd ZZS)LoweringExceptionNexc Exceptionr rrrrr stack_trace str | Noner r c sJt|jd|d||||}|r||d|7}t|dS)N: rz Found from : )typer*r)r0r1)r3r=r rrr?msgr4rrr1@s$zLoweringException.__init__r) r=r>r rrrrrr?r@r r r7rrr4rr<?sr<c@ eZdZdS)SubgraphLoweringExceptionNr*r+r,rrrrrENrEcseZdZdfdd ZZS)InvalidCxxCompilerr r cs.ddlm}td|jd|jjdS)N)configz!No working C++ compiler found in z .cpp.cxx: )rJr0r1r*cppcxx)r3rJr4rrr1Ss zInvalidCxxCompiler.__init__)r r r7rrr4rrHRr9rHceZdZdfdd ZZS)CppWrapperCodegenErrorrCrr r cstd|dS)NzC++ wrapper codegen error: r0r1)r3rCr4rrr1\szCppWrapperCodegenError.__init__)rCrr r r7rrr4rrO[r9rOcs(eZdZd fdd Zd d d ZZS)CppCompileErrorcmd list[str]outputrr r csJt|tr |d}||_||_ttd j d ||ddS)Nzutf-8z C++ compile error Command: {cmd} Output: {output}  )rRrT) isinstancebytesdecoderRrTr0r1r%r;stripformatr')r3rRrTr4rrr1as   zCppCompileError.__init__"tuple[type, tuple[list[str], str]]cCs|j|j|jffSr)r5rRrT)r3rrr __reduce__xszCppCompileError.__reduce__)rRrSrTrr r )r r[)r*r+r,r1r\r8rrr4rrQ`srQc@rD)CUDACompileErrorNrFrrrrr]|rGr]crN) TritonMissingfirst_useful_frametypes.FrameType | Noner 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/tritonr_rP)r3r_r4rrr1s zTritonMissing.__init__)r_r`r r r7rrr4rr^r9r^cseZdZd fdd ZZS) GPUTooOldForTriton device_propsrr_r`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 .ra)r0r1namemajorminor)r3rcr_r4rrr1s  zGPUTooOldForTriton.__init__)rcrr_r`r r r7rrr4rrbr9rbc@seZdZdZd dd Zd S) InductorErrorinductorinner_exceptionr>r_r`r r cCs*||_tj|t|jd||ddS)NrAra)rjrr1rBr*)r3rjr_rrrr1s  zInductorError.__init__N)rjr>r_r`r r )r*r+r, backend_namer1rrrrrhsrh)r rr r ) __future__rosrr% functoolsrtypingrrtorch._dynamo.excrrtypes torch.cudarenvirongetr RuntimeErrorrr/r:r<rErHrOrQr]r^rbrhrrrrs4