polars.lazyframe.engine_config.GPUEngine

class polars.lazyframe.engine_config.GPUEngine(
    *,
    device: int | None = None,
    memory_resource: Any | None = None,
    raise_on_fail: bool = False,
    monitoring: bool = False,
    **kwargs: Any,
)

Параметры конфигурации движка выполнения на GPU.

Используйте этот класс, если хотите управлять деталями выполнения.

Параметры:

deviceint, default None

Выберите GPU для выполнения запроса. Если устройство не указано, запрос выполняется на текущем устройстве CUDA.

memory_resourcermm.mr.DeviceMemoryResource, default None

Укажите ресурс памяти для выделения памяти GPU.

Предупреждение

Если передаётся memory_resource, необходимо убедиться, что он подходит для выбранного device. Подробнее см. в документации RMM.

raise_on_failbool, default False

Если значение равно True, не переключаться на движок Polars для CPU, если движок GPU не может выполнить запрос, а вместо этого вызывать ошибку.

__init__(
    *,
    device: int | None = None,
    memory_resource: Any | None = None,
    raise_on_fail: bool = False,
    monitoring: bool = False,
    **kwargs: Any,
) → None

Методы

__init__(*[, device, memory_resource, ...])
collect(lf, *, optimizations[, background, ...])	Выполняет lf и возвращает DataFrame или дескриптор фонового запроса.
collect_all(lfs, *, optimizations)	Выполняет несколько запросов, при возможности параллельно.
collect_all_async(lfs, *, optimizations[, ...])	Асинхронно выполняет несколько запросов.
collect_async(lf, *, optimizations[, gevent])	Асинхронно выполняет lf.
collect_batches(lf, *, optimizations[, ...])	Выполняет lf, выдавая результат пакетами.
execute(lf, *, optimizations)	Выполняет запрос в QueryResult.
sink_batches(lf, function, *, chunk_size, ...)	См. polars.LazyFrame.sink_batches().
sink_csv(lf, path, *, include_bom, ...)	См. polars.LazyFrame.sink_csv().
sink_ipc(lf, path, *, compression, ...)	См. polars.LazyFrame.sink_ipc().
sink_ndjson(lf, path, *, compression, ...)	См. polars.LazyFrame.sink_ndjson().
sink_parquet(lf, path, *, compression, ...)	См. polars.LazyFrame.sink_parquet().

Атрибуты

monitoring	Отслеживаются ли запросы (None использует настроенное значение по умолчанию).
name	Название движка.
plan_engine	Название движка, используемое при формировании планов запросов.
device	Устройство для выполнения запроса.
memory_resource	Ресурс памяти для выделения памяти устройства.
raise_on_fail	Следует ли вызывать ошибку для неподдерживаемых запросов вместо переключения на движок CPU.
config	Дополнительные параметры конфигурации движка.