Skip to content

Make masking more efficient for reduce_to_vector #14

Open
@jim22k

Description

@jim22k

The mask in reduce_to_vector is a Vector, while the input is a Matrix, making it impossible to apply the mask prior to the operation. If select_by_mask could do vec->matrix broadcasting, the mask could be applied earlier and possibly save compute time.

Alternatively, the mask could be brought into the linalg.generic loop and used to skip entire rows.

The 2nd approach would be faster and simpler if it works.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions