已关闭
[Feature]: HyperParallel-FSDP MS后端支持CPUOffload #416
mindspore-ci-bot创建于  9月20日关闭于  9月20日
mindspore-ci-bot成员
9月20日 创建

Checklist

🐛 Describe the bug

HyperParallel-FSDP在使用CPUOffload时,报错参数切片不在CPU上。然而期望参数切片(即sharded_param存在与CPU上)
问题在于当前Mindspore框架默认行为是CPU上的Tensor通过计算算子,会隐式的搬运到NPU上。经过和同事线下讨论。该问题依赖框架机制,因此该问题通过该需求跟踪,待讨论清楚框架机制后在MS后端支持CPUOffload特性。

Expected behavior

期望行为是:CPUOffload入参用来控制是否将参数,梯度,优化器状态的分片放置在CPU上,从而进一步减少Device上的显存占用。最后优化器更新会在CPU上进行.
fully_shard内部控制参数切片,梯度切片卸载到CPU上。包括对梯度累加场景是对梯度累加的+=都放置在CPU上进行。优化器动量通过在CPUDeveiceContext下创建,动量也可以放到CPU上。最终优化器.step在CPU上进行。

Additional context

Environment info

Mindspore:2.10.0
HyperParallel: r1.0.0
Cann 9.0.0

Thanks for contributing 🎉!

schema_version: 1
source: gitcode
gitcode_repo: mindspore/hyper-parallel
gitcode_issue: 249
source_url: https://gitcode.com/mindspore/hyper-parallel/issues/249

likedislike
Mmindspore-ci-bot成员
9月20日 修改了issue 的描述
Mmindspore-ci-bot成员
9月20日 关闭了 issue
mindspore-ci-bot成员
9月20日 评论:

GitHub issue mindspore-ai/hyper-parallel#242 was closed.

likedislike
MindSpore-BotMindSpore-Bot成员
9月20日 issue状态由 TODO 改变为 DONE