DeepSpeed
Fix crash when creating Torch tensor on NPU with device=get_accelerator().current_device()
#5464
Merged

Fix crash when creating Torch tensor on NPU with device=get_accelerator().current_device() #5464

loadams merged 4 commits into deepspeedai:master from harygo2:master
harygo2
harygo2 replacing current_device with current_device_name to support other ac…
2f382d89
harygo2 harygo2 requested a review from mrwyattii mrwyattii 1 year ago
harygo2 harygo2 requested a review from tjruwase tjruwase 1 year ago
harygo2 harygo2 requested a review from loadams loadams 1 year ago
harygo2
minchao-sun
tjruwase
tjruwase
tjruwase approved these changes on 2024-04-26
tjruwase Merge branch 'master' into master
1e5e1b4c
harygo2 Update utils.py
2eadf2fe
harygo2
harygo2
delock
harygo2
harygo2
loadams
loadams Merge branch 'master' into master
d8bb1fd5
loadams loadams enabled auto-merge 1 year ago
loadams loadams merged 0fc19b6a into master 1 year ago

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone