o
    in                     @   sv   U d dl mZmZ ddlmZmZ i Zeeeege	f f e
d< dedeege	f ddfd	d
Zdede	fddZdS )    )AnyCallable   )*get_minimal_required_cuda_ver_from_ptx_verget_ptx_ver_handle_getterstgetterreturnNc                 C   s   |t | < d S )N)r   )r   r	    r   Y/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/cuda/bindings/utils/__init__.py_add_cuda_native_handle_getter
   s   r   objc                 C   s8   t | }zt| | W S  ty   tdt| dw )zReturns the address of the provided CUDA Python object as a Python int.

    Parameters
    ----------
    obj : Any
        CUDA Python object

    Returns
    -------
    int : The object address.
    zUnknown type: N)typer   KeyError	TypeErrorstr)r   obj_typer   r   r   get_cuda_native_handle   s   r   )typingr   r   
_ptx_utilsr   r   r   dictr   int__annotations__r   r   r   r   r   r   <module>   s
    