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

run PyTorch GC for ROCm devices #323

Closed
ssube opened this issue Apr 11, 2023 · 0 comments
Closed

run PyTorch GC for ROCm devices #323

ssube opened this issue Apr 11, 2023 · 0 comments
Labels
provider/rocm scope/api status/fixed issues that have been fixed and released type/feature new features
Milestone
v0.10

Comments

@ssube
Copy link
Owner

ssube commented Apr 11, 2023 •
edited

PyTorch uses the same API and naming for CUDA and ROCm devices, so it should be possible to run the Torch device GC on AMD as well (for Linux, at least).
ssube added a commit that referenced this issue Apr 11, 2023
@ssube
fix(api): run PyTorch GC on ROCm devices (#323)
00be4f4
@ssube ssube added status/fixed issues that have been fixed and released type/feature new features scope/api provider/rocm labels Apr 11, 2023
@ssube ssube added this to the v0.10 milestone Apr 11, 2023
@ssube ssube closed this as completed Apr 12, 2023
@ssube ssube mentioned this issue May 1, 2023
Closed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
provider/rocm scope/api status/fixed issues that have been fixed and released type/feature new features
Projects
None yet
Milestone
v0.10
Development

No branches or pull requests
1 participant
@ssube