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

[ROCm] enable cupy in order to enable cudagraph mode for AMD GPUs #3123

Merged
merged 4 commits into from
Mar 5, 2024
Merged

[ROCm] enable cupy in order to enable cudagraph mode for AMD GPUs #3123

merged 4 commits into from
Mar 5, 2024

Commits on Feb 29, 2024

  1. [ROCm] enable cupy for AMD GPUs

    @hongxiayang
    hongxiayang committed Feb 29, 2024
    Configuration menu
    Copy the full SHA
    7a921e6 View commit details
    Browse the repository at this point in the history

Commits on Mar 1, 2024

  1. update cupy branch and enable cupy for rocm

    @lcskrishna
    lcskrishna committed Mar 1, 2024
    Configuration menu
    Copy the full SHA
    39e672b View commit details
    Browse the repository at this point in the history

  2. Merge pull request #1 from lcskrishna/vllm-cupy-pr 

    Update cupy branch and enable CuPy for allreduce.
    @hongxiayang
    hongxiayang authored Mar 1, 2024
    Configuration menu
    Copy the full SHA
    ebb8ec3 View commit details
    Browse the repository at this point in the history

  3. fix cupy installation problem and ruff

    @hongxiayang
    hongxiayang committed Mar 1, 2024
    Configuration menu
    Copy the full SHA
    43c606c View commit details
    Browse the repository at this point in the history