PaddlePaddle / PaddlePaddle/Paddle
【CustomDevice】CustomDevice在调用flatten_dense_tensors报错
Open
@changeyoung98 is already working on this.
Since Nov 10, 2023.
status/developing
type/bug-report
- Dominant language
- C++
- Stars
- 24.1k
- Forks
- 6k
- Avg merge
- 4d 17h
- Merged PRs (30d)
- 60
Description
bug描述 Describe the Bug
目前FusedCommBuffer功能会创建ParamStorage以及GradStorage，但是目前flatten_dense_tensors函数只支持了GPU没有支持CustomDevice，固定写了device为‘GPU’，预期希望flatten_dense_tensors支持该功能。
其他补充信息 Additional Supplementary Information
No response
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Assessment
This issue has not been assessed yet.