Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

GPU support with CuPy #37

Open
wants to merge 19 commits into
base: master
Choose a base branch
from
Open

GPU support with CuPy #37

wants to merge 19 commits into from

Commits on Dec 8, 2023

  1. Implemented CuPy version of DistArray

    Thomas committed Dec 8, 2023
    Configuration menu
    Copy the full SHA
    a50137f View commit details
    Browse the repository at this point in the history

  2. Cleanup and tests

    Thomas committed Dec 8, 2023
    Configuration menu
    Copy the full SHA
    e770b56 View commit details
    Browse the repository at this point in the history

Commits on Dec 12, 2023

  1. Added cupy backend for FFTs

    Thomas committed Dec 12, 2023
    Configuration menu
    Copy the full SHA
    424fdb1 View commit details
    Browse the repository at this point in the history

  2. Removed useCuPy argument from NewDistArray

    Thomas committed Dec 12, 2023
    Configuration menu
    Copy the full SHA
    44be7bc View commit details
    Browse the repository at this point in the history

  3. Fixed tests

    Thomas committed Dec 12, 2023
    Configuration menu
    Copy the full SHA
    90551b4 View commit details
    Browse the repository at this point in the history

Commits on Dec 14, 2023

  1. Added cupyx-scipy backend

    Thomas committed Dec 14, 2023
    Configuration menu
    Copy the full SHA
    202f30c View commit details
    Browse the repository at this point in the history

Commits on Dec 18, 2023

  1. Implemented NCCL for Alltoallw operations

    Thomas committed Dec 18, 2023
    Configuration menu
    Copy the full SHA
    e5d4b56 View commit details
    Browse the repository at this point in the history

  2. Minor refactor

    Thomas committed Dec 18, 2023
    Configuration menu
    Copy the full SHA
    d0a493e View commit details
    Browse the repository at this point in the history

Commits on Dec 20, 2023

  1. Changed Alltoallw communication pattern

    Thomas committed Dec 20, 2023
    Configuration menu
    Copy the full SHA
    6b47e36 View commit details
    Browse the repository at this point in the history

Commits on Jan 4, 2024

  1. Added custom MPI based Alltoallw implementation

    Thomas committed Jan 4, 2024
    Configuration menu
    Copy the full SHA
    21805de View commit details
    Browse the repository at this point in the history

Commits on Jan 12, 2024

  1. Cleanup

    Thomas committed Jan 12, 2024
    Configuration menu
    Copy the full SHA
    ceccc7d View commit details
    Browse the repository at this point in the history

Commits on Feb 13, 2024

  1. Overlapped all communication within a single Alltoallw in NCCL backend

    Thomas committed Feb 13, 2024
    Configuration menu
    Copy the full SHA
    c3c20ff View commit details
    Browse the repository at this point in the history

  2. Cosmetic changes

    Thomas committed Feb 13, 2024
    Configuration menu
    Copy the full SHA
    405fb4d View commit details
    Browse the repository at this point in the history

Commits on Feb 23, 2024

  1. Swapped array[...]=data for cupy.copyto

    Thomas committed Feb 23, 2024
    Configuration menu
    Copy the full SHA
    b69a303 View commit details
    Browse the repository at this point in the history

Commits on Feb 27, 2024

  1. Removed unnecessary rescaling in CuPy backend

    Thomas committed Feb 27, 2024
    Configuration menu
    Copy the full SHA
    b9ac0f7 View commit details
    Browse the repository at this point in the history

Commits on Feb 29, 2024

  1. Use CUDA graphs in NCCL Alltoallw

    Thomas committed Feb 29, 2024
    Configuration menu
    Copy the full SHA
    77b96f3 View commit details
    Browse the repository at this point in the history

Commits on Apr 16, 2024

  1. Removed dependence on CuPy

    Thomas committed Apr 16, 2024
    Configuration menu
    Copy the full SHA
    75f0343 View commit details
    Browse the repository at this point in the history

Commits on May 24, 2024

  1. Merge remote-tracking branch 'upstream/master' into cupy_implementation

    Thomas committed May 24, 2024
    Configuration menu
    Copy the full SHA
    a920dca View commit details
    Browse the repository at this point in the history

Commits on Oct 12, 2024

  1. Fixes

    Thomas committed Oct 12, 2024
    Configuration menu
    Copy the full SHA
    7330b6b View commit details
    Browse the repository at this point in the history