DeepSpeed
[BUG FIX]:fix get torch.version.cuda error when cuda is None in rocm
#6909
Merged
Merged
[BUG FIX]:fix get torch.version.cuda error when cuda is None in rocm #6909
loadams
merged
7 commits into
deepspeedai:master
from
kairos-yu:dev_hjwei
[BUG FIX]:fix get torch.version.cuda error when cuda is None in rocm
b86b3804
tjruwase
commented
on 2024-12-24
Move ROCm torch branch check into get_cuda_compile_flag
309a666d
Merge branch 'master' into dev_hjwei
1406adc6
Merge branch 'master' into dev_hjwei
fb4a6d1a
loadams
approved these changes
on 2024-12-30
Merge branch 'master' into dev_hjwei
2b66d009
Merge branch 'master' into dev_hjwei
a98615fb
Merge branch 'master' into dev_hjwei
1b7f49d4
Login to write a write a comment.
Login via GitHub