Skip to content

Use ncclAvg reduce op instead of future hook that divide tensor by world size #34

Description

@gpauloski

Metadata

Metadata

Assignees

No one assigned

    Labels

    enhancementNew feature or requesthelp wantedExtra attention is neededpytorch-1.11Features available in PyTorch 1.11

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions