o ¹­§i…ã@sJddlmZddlmZmZerddlmZddlmZGdd„dƒZ dS)é)Ú annotations)Ú TYPE_CHECKINGÚAny)ÚMapping)ÚDeviceMemoryResourcec@sNeZdZUdZded< ded< ded< ded < d d d d œddd„Zd 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 `_ 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ÚdevicezDeviceMemoryResource | NoneÚmemory_resourceÚboolÚ raise_on_failzMapping[str, Any]ÚconfigNF)r r r ú Any | NoneÚkwargsrÚreturnÚNonecKs||_||_||d<||_dS)Nr )r r r )Úselfr r r r©rúU/home/app/Keep/.python/lib/python3.10/site-packages/polars/lazyframe/engine_config.pyÚ__init__1s zGPUEngine.__init__) r rr rr r rrrr)Ú__name__Ú __module__Ú __qualname__Ú__doc__Ú__annotations__rrrrrr s ûrN) Ú __future__rÚtypingrrÚcollections.abcrZrmm.mrrrrrrrÚs