issues
search
hpcaitech
/
ColossalAI
Making large AI models cheaper, faster and more accessible
https://www.colossalai.org
Apache License 2.0
38.68k
stars
4.34k
forks
source link
[hotfix] Fix object_to_tensor usage when torch>=2.3.0
#5820
Closed
kurisusnowdeng
closed
2 months ago
kurisusnowdeng
commented
3 months ago
📝 What does this PR do?
Pytorch 2.3.0 added a
group
argument in
_object_to_tensor
function. Updated related use cases in pipeline communication.
📝 What does this PR do?
group
argument in_object_to_tensor
function. Updated related use cases in pipeline communication.