
    q-Ph                    X    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	dS )    )annotations)TYPE_CHECKINGAny)Mapping)DeviceMemoryResourcec                  V    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d
S )	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   
Any | Nonekwargsr   returnNonec               :    || _         || _        ||d<   || _        d S )Nr   )r   r   r   )selfr   r   r   r   s        ^/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/polars/lazyframe/engine_config.py__init__zGPUEngine.__init__1   s(     ."/    )
r   r
   r   r   r   r   r   r   r   r   )__name__
__module____qualname____doc____annotations__r    r   r   r	   r	      s          2 '00008 :
 "&*#       r   r	   N)

__future__r   typingr   r   collections.abcr   rmm.mrr   r	   r   r   r   <module>r#      s    " " " " " " % % % % % % % % ,''''''++++++2 2 2 2 2 2 2 2 2 2r   