Skip to content

[FEA] Allow RMM_DEFAULT_HOST_ALIGNMENT alignment in pool_memory_resource #1585

Open
@PointKernel

Description

@PointKernel

Is your feature request related to a problem? Please describe.
Currently, the alignment check in pool_memory_resource ctor is hardcoded to check against CUDA_ALLOCATION_ALIGNMENT:

RMM_EXPECTS(rmm::is_aligned(initial_pool_size, rmm::CUDA_ALLOCATION_ALIGNMENT),
"Error, Initial pool size required to be a multiple of 256 bytes");
RMM_EXPECTS(rmm::is_aligned(maximum_pool_size.value_or(0), rmm::CUDA_ALLOCATION_ALIGNMENT),
"Error, Maximum pool size required to be a multiple of 256 bytes");

This causes problems for host memory pools.

Additional context
Related issue: rapidsai/cudf#16024

Metadata

Metadata

Assignees

No one assigned

    Labels

    feature requestNew feature or requestwontfixThis will not be worked on

    Type

    No type

    Projects

    Status

    Blocked

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions