Ë
    	êñi…  ã                  óP   — d dl mZ d dlmZmZ erd dlmZ d dlmZ  G d„ d«      Z	y)é    )Úannotations)ÚTYPE_CHECKINGÚAny)ÚMapping)ÚDeviceMemoryResourcec                  óf   — e Zd ZU dZded<   	 ded<   	 ded<   	 ded	<   	 d
d
ddœ	 	 	 	 	 	 	 	 	 dd„Zy
)Ú	GPUEngineaq  
    Configuration options for the GPU execution engine.

    Use this if you want control over details of the execution.

    Parameters
    ----------
    device : int, default None
        Select the GPU used to run the query. If not provided, the
        query uses the current CUDA device.
    memory_resource : rmm.mr.DeviceMemoryResource, default None
        Provide a memory resource for GPU memory allocations.

        .. warning::
           If passing a `memory_resource`, you must ensure that it is valid
           for the selected `device`. See the `RMM documentation
           <https://github.com/rapidsai/rmm?tab=readme-ov-file#multiple-devices>`_
           for more details.

    raise_on_fail : bool, default False
        If True, do not fall back to the Polars CPU engine if the GPU
        engine cannot execute the query, but instead raise an error.

    ú
int | NoneÚdevicezDeviceMemoryResource | NoneÚmemory_resourceÚboolÚraise_on_failzMapping[str, Any]ÚconfigNF)r   r   r   c               ó8   — || _         || _        ||d<   || _        y )Nr   )r   r   r   )Úselfr   r   r   Úkwargss        ú`/var/www/pod-logistic/pod-ai/venv/lib/python3.12/site-packages/polars/lazyframe/engine_config.pyÚ__init__zGPUEngine.__init__1   s$   € ð ˆŒØ.ˆÔà"/ˆˆÑØˆ�ó    )
r   r
   r   z
Any | Noner   r   r   r   ÚreturnÚNone)Ú__name__Ú
__module__Ú__qualname__Ú__doc__Ú__annotations__r   © r   r   r	   r	      ss   … ñð2 ÓØ'Ø0Ó0Ø8ØÓðð ÓØ:ð
 "Ø&*Ø#ñð ðð $ð	ð
 ðð ðð 
ôr   r	   N)
Ú
__future__r   Útypingr   r   Úcollections.abcr   Úrmm.mrr   r	   r   r   r   ú<module>r"      s   ðÝ "ç %áÝ'å+÷2ò 2r   