-i6UddlmZmZddlmZGddeZGddeZGddZeZed e Z eee d <d ed d fdZ GddZ e eZ d eed d fdZy ))OptionalProtocol) ContextVarceZdZdefdZy)BufferreturncyNselfs d/home/obispo/Crisostomo_bridge/mision_env/lib/python3.12/site-packages/triton/runtime/_allocation.pydata_ptrzBuffer.data_ptr N)__name__ __module__ __qualname__intrr rrrrs # rrc*eZdZdededeedefdZy) Allocatorsize alignmentstreamrcyr r r rrrs r__call__zAllocator.__call__ rrNrrrrrrrr rrrr s' S S (3- F rrc*eZdZdededeedefdZy) NullAllocatorrrrrctd)NzxKernel requires a runtime memory allocation, but no allocator was set. Use triton.set_allocator to specify an allocator.) RuntimeErrorrs rrzNullAllocator.__call__sOP PrNrr rrr r s,PSPSP(3-PFPrr _allocator)default allocatorrNc.tj|y)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 NN9rcZeZdZdZdeddfdZdefdZdeddfdZded ed e ede fd Z y) _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||_yr r#r r%s r__init__z_AllocatorWrapper.__init__+ #rc|jSr r-r s rgetz_AllocatorWrapper.get.s rc||_yr r-r.s rr'z_AllocatorWrapper.set1r0rrrrc(|j|||Sr r-rs rrz_AllocatorWrapper.__call__4stY77r) rrr__doc__rr/r2r'rrrrr rrr+r+%s^ $)$$Y$Y$4$8S8S8(3-8F8rr+cHtj||yty)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  (=9S?Sr)typingrr contextvarsrrrr r8r#__annotations__r)r+r7r9r rrr=s%%" X   PP /$.|_$U Jy !UY488&'7TXi%8TTTr