o ^i@sUddlmZmZddlmZGdddeZGdddeZGdddZed ed Zeee d <d efd d Z ed ed a ee d<d eefddZ dS))OptionalProtocol) ContextVarc@seZdZdefddZdS)BufferreturncCdSN)selfr r U/var/www/html/RAG/RAG_venv/lib/python3.10/site-packages/triton/runtime/_allocation.pydata_ptrzBuffer.data_ptrN)__name__ __module__ __qualname__intr r r r r rsrc@*eZdZdededeedefddZdS) Allocatorsize alignmentstreamrcCrrr r rrrr r r __call__ r zAllocator.__call__Nrrrrrrrr r r r r "rc@r) NullAllocatorrrrrcCstd)NzxKernel requires a runtime memory allocation, but no allocator was set. Use triton.set_allocator to specify an allocator.) RuntimeErrorrr r r rszNullAllocator.__call__Nrr r r r rrr _allocator)default allocatorcCt|dS)z The allocator function is called during kernel launch for kernels that require additional global memory workspace. N)rsetrr r r set_allocatorsr#_profile_allocatorcCr )z The profile allocator function is called before kernel launch for kernels that require additional global memory workspace. N)r$r!r"r r r set_profile_allocator&sr%N) typingrr contextvarsrrrrr__annotations__r#r$r%r r r r s