DeepSpeed
Add HIP device abstraction, update Triton skip logic
#5120
Open

Add HIP device abstraction, update Triton skip logic #5120

lekurile wants to merge 9 commits into master from lekurile/add_hip_abstraction
lekurile
lekurile Add HIP device abstraction, update Triton skip logic
0a72e6f9
lekurile lekurile requested a review from tjruwase tjruwase 1 year ago
lekurile lekurile requested a review from awan-10 awan-10 1 year ago
lekurile lekurile requested a review from loadams loadams 1 year ago
lekurile lekurile requested a review from mrwyattii mrwyattii 1 year ago
loadams
lekurile Add try-except to hip accelerator
feda27e5
loadams
loadams approved these changes on 2024-02-12
mrwyattii
mrwyattii commented on 2024-02-12
lekurile Clean up HIP accelerator
943cd82e
lekurile import torch in HIP accelerator check
09ee6e11
loadams Merge branch 'master' into lekurile/add_hip_abstraction
a80e0a5c
lekurile Merge branch 'master' into lekurile/add_hip_abstraction
050a45d4
loadams
loadams Merge branch 'master' into lekurile/add_hip_abstraction
8bf2aa99
lekurile Merge branch 'lekurile/add_hip_abstraction' of https://github.com/mic…
d2734ce4
lekurile Merge branch 'master' into lekurile/add_hip_abstraction
7d42638b

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone