Otj6UddlmZmZddlmZGddeZGddeZGddZeZed e Z eee d <d ed d fdZ GddZ e eZ d eed d fdZd S))OptionalProtocol) ContextVarceZdZdefdZdS)BufferreturncdSNselfs X/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/triton/runtime/_allocation.pydata_ptrzBuffer.data_ptr N)__name__ __module__ __qualname__intrr rrrrs/ #      rrc2eZdZdededeedefdZdS) Allocatorsize alignmentstreamrcdSr r r rrrs r__call__zAllocator.__call__ rrNrrrrrrrr rrrr 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 rrzNullAllocator.__call__sOPP PrNrr rrr r sRPSPSP(3-PFPPPPPPrr _allocator)default allocatorrNc: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 NN9rcbeZdZdZdeddfdZdefdZdeddfdZded ed e ede fd Z dS) _AllocatorWrapperz Wrapper to provide ContextVar-like .get()/.set() methods. profile_allocator is used in same way as allocator so it is useful to maintain the interface. r%rNc||_dSr r#r r%s r__init__z_AllocatorWrapper.__init__+ #rc|jSr r-r s rgetz_AllocatorWrapper.get.s rc||_dSr r-r.s rr'z_AllocatorWrapper.set1r0rrrrc0||||Sr r-rs rrz_AllocatorWrapper.__call__4stY777r) rrr__doc__rr/r2r'rrrrr rrr+r+%s $)$$$$$Y$Y$4$$$$8S8S8(3-8F888888rr+cLt||ntdS)z The profile allocator function is called before kernel launch for kernels that require additional global memory workspace. N)_profile_allocatorr'_NULL_ALLOCATORr(s rset_profile_allocatorr9;s'  (=99?SSSSSr)typingrr contextvarsrrrr r8r#__annotations__r)r+r7r9r rrr=s}%%%%%%%%%""""""     X           PPPPPPPP -//$.J|_$U$U$U Jy !UUUY488888888&'&77TXi%8TTTTTTTTr