Skip to content

asarray(masked=True) does not work with NCCL #140

@mrava87

Description

@mrava87

PR #139 introduces the masked parameter to asarray to be able to collect a masked array from all ranks where the mask is the one given when initializing the DistributedArray.

@tharittk discovered an issue with the NCCL backend likely due to self.local_shapes (which call all GPUs in the world communicator). MPI does not have this problem because it sends the object and no buffer / array unrolling needed.

Metadata

Metadata

Assignees

No one assigned

    Labels

    bugSomething isn't working

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions