Skip to content

Allow stack operations to be fused #414

Open
@tomwhite

Description

@tomwhite

The way stack is currently implemented is slightly different to the other functions in that the number of arguments passed to the blockwise function _read_stack_chunk is just one (the chunk being stored), which is different to the number of arrays that the stack operation in the DAG has as predecessor nodes (the arrays being stacked).

This means that the fusion logic in blockwise breaks in this case. A simple workaround is to mark the stack operation as not fusable, but it would be better to fix this by removing the limitation directly.

Metadata

Metadata

Assignees

No one assigned

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions