PiUddlmZmZddlmZGddeZGddeZGddZed e Zeee d <d efd Z ed e a ee d <d eefdZ dS))OptionalProtocol) ContextVarceZdZdefdZdS)BufferreturncdSN)selfs r/var/www/development/aibuddy-work/election-extract/venv/lib/python3.11/site-packages/triton/runtime/_allocation.pydata_ptrzBuffer.data_ptr N)__name__ __module__ __qualname__intrr rr rrs/ #      rrc2eZdZdededeedefdZdS) Allocatorsize alignmentstreamrcdSr r r rrrs r __call__zAllocator.__call__ rrNrrrrrrrr rr rr sH S S (3- F      rrc2eZdZdededeedefdZdS) NullAllocatorrrrrc td)NzxKernel requires a runtime memory allocation, but no allocator was set. Use triton.set_allocator to specify an allocator.) RuntimeErrorrs r rzNullAllocator.__call__sOPP PrNrr rr rrsRPSPSP(3-PFPPPPPPrr _allocator)default allocatorc:t|dS)z The allocator function is called during kernel launch for kernels that require additional global memory workspace. N)r"setr$s r set_allocatorr(s NN9r_profile_allocatorc:t|dS)z The profile allocator function is called before kernel launch for kernels that require additional global memory workspace. N)r)r&r's r set_profile_allocatorr+&s 9%%%%%rN) typingrr contextvarsrrrrr"__annotations__r(r)r+r rr r/sO%%%%%%%%%""""""     X           PPPPPPPP%/J|]]__$U$U$U Jy !UUUY!+ < Q Q QIQQQ&Xi%8&&&&&&r