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

Fix search path for torch allocator in editable installs and ensure CUDA support is available #1498

Merged
merged 5 commits into from
Mar 8, 2024
Merged

Fix search path for torch allocator in editable installs and ensure CUDA support is available #1498

merged 5 commits into from
Mar 8, 2024

Commits on Mar 5, 2024

  1. Fix search path for torch allocator in editable installs

    @vyasr
    vyasr committed Mar 5, 2024
    Configuration menu
    Copy the full SHA
    d913a8a View commit details
    Browse the repository at this point in the history

  2. Verify that the torch version we have is GPU-enabled before setting t… 

    …he allocator
    @vyasr
    vyasr committed Mar 5, 2024
    Configuration menu
    Copy the full SHA
    75fefec View commit details
    Browse the repository at this point in the history

Commits on Mar 6, 2024

  1. Merge branch 'branch-24.04' into fix/torch_allocator_editable

    @bdice
    bdice committed Mar 6, 2024
    Configuration menu
    Copy the full SHA
    dda47cd View commit details
    Browse the repository at this point in the history

Commits on Mar 8, 2024

  1. Update python/rmm/allocators/torch.py 

    Co-authored-by: Mark Harris <783069+harrism@users.noreply.github.com>
    @vyasr @harrism
    vyasr and harrism committed Mar 8, 2024
    Configuration menu
    Copy the full SHA
    049544e View commit details
    Browse the repository at this point in the history

  2. Merge branch 'branch-24.04' into fix/torch_allocator_editable

    @vyasr
    vyasr committed Mar 8, 2024
    Configuration menu
    Copy the full SHA
    bcf63b4 View commit details
    Browse the repository at this point in the history