Skip to content

Investigate alternative partitioning in tridiagonal solver #790

@msimberg

Description

@msimberg

// The number of non-deflated values
k = n - thrust::count(par, c_ptr, c_ptr + n, ColType::Deflated);
// Partition while preserving relative order such that deflated entries are at the end
thrust::stable_partition_copy(par, in_ptr, in_ptr + n, out_ptr, out_ptr + k,
PartitionIndicesPredicate{c_ptr});
could potentially be done using a copy and then in-place partition instead. Investigate if it's faster than the current method.

This is low priority unless we find evidence that this kernel is a bottleneck.

Metadata

Metadata

Assignees

No one assigned

    Type

    Projects

    Status

    No status

    Milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions