
    j                         U d dl mZmZ ddlmZmZ ddl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
ZdedefdZd	S )    )AnyCallable   )*get_minimal_required_cuda_ver_from_ptx_verget_ptx_ver)#warn_if_cuda_major_version_mismatch_handle_getterstgetterreturnNc                     |t           | <   d S )N)r	   )r
   r   s     a/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/cuda/bindings/utils/__init__.py_add_cuda_native_handle_getterr      s    OA    objc                     t          |           }	 t          |         |           S # t          $ r! t          dt	          |          z             dw xY 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_types     r   get_cuda_native_handler      sc     CyyHDx(--- D D D(3x==899tCDs	   & +A)typingr   r   
_ptx_utilsr   r   _version_checkr   r	   dictr   int__annotations__r   r    r   r   <module>r       s    !                 O O O O O O O O ? ? ? ? ? ?46dHcUCZ001 6 6 6 d  HcUCZ4H  T        D D D D D D D Dr   