Skip to content

Conversation

@zty-king
Copy link
Contributor

PR Category

Operator Mechanism

PR Types

Improvements

Description

当前load_state_dict使用fc框架时,只支持多卡或单卡策略转多卡策略,未支持多卡策略转单卡策略,加以补充,一方面用于单机合参,另一方面用于多卡策略转单机训练或推理

@paddle-bot
Copy link

paddle-bot bot commented Sep 20, 2025

你的PR提交成功,感谢你对开源项目的贡献!
请关注后续CI自动化测试结果,详情请参考Paddle-CI手册
Your PR has been submitted. Thanks for your contribution!
Please wait for the result of CI firstly. See Paddle CI Manual for details.

@paddle-bot paddle-bot bot added the contributor External developers label Sep 20, 2025
@zty-king
Copy link
Contributor Author

/re-run all-failed

2 similar comments
@zty-king
Copy link
Contributor Author

/re-run all-failed

@zty-king
Copy link
Contributor Author

/re-run all-failed

@xingmingyyj
Copy link
Contributor

LGTM

@xingmingyyj
Copy link
Contributor

/re-run all-failed

2 similar comments
@zty-king
Copy link
Contributor Author

/re-run all-failed

@xingmingyyj
Copy link
Contributor

/re-run all-failed

Copy link
Contributor

@From00 From00 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@xingmingyyj
Copy link
Contributor

/re-run all-failed

5 similar comments
@zty-king
Copy link
Contributor Author

/re-run all-failed

@zty-king
Copy link
Contributor Author

/re-run all-failed

@xingmingyyj
Copy link
Contributor

/re-run all-failed

@xingmingyyj
Copy link
Contributor

/re-run all-failed

@zty-king
Copy link
Contributor Author

/re-run all-failed

@xuxinyi389 xuxinyi389 merged commit f7bcd42 into PaddlePaddle:develop Sep 28, 2025
168 of 181 checks passed
@zty-king zty-king changed the title handle_missing_edge_cases_in_fc 【FlexCheckpoint】handle_missing_edge_cases_in_fc Oct 11, 2025
xingmingyyj pushed a commit to xingmingyyj/Paddle that referenced this pull request Oct 22, 2025
swgu98 pushed a commit that referenced this pull request Oct 23, 2025
…#75996)

* [Flex CP]Fix merge_sharded_state_dict with aoa and offload (#75062)

* fix merge_state_dict with aoa and offload

* add tests

* refine

* fix

* fix

* add log

* fix

* fix

* 【FlexCheckpoint】Upgrade some macros and optimize load_state_dict communication (#75282)

* upgrad macros and load_state_dict comm task

fix

fix

support 0-d tensor

fix

balance save and fix

* fix test

* Add the test about the sharded_state_dict of optimizer  (#75067)

* fix the share_weight_bug

* add note

* add the unit test

* set the timeout

* add more test

* Trigger CI rebuild

* fix the CmakeLists

* handle_missing_edge_cases_in_fc (#75413)

* up_grade fc (#75613)

fix and add test

fix

fix

fix

fix cmakelists

add notion

---------

Co-authored-by: Chen Zhiyang <[email protected]>
Co-authored-by: Tianyu Zheng <[email protected]>
xingmingyyj pushed a commit to xingmingyyj/Paddle that referenced this pull request Nov 5, 2025
sneaxiy pushed a commit that referenced this pull request Nov 6, 2025
….2 (#76249)

* 【FlexCP】merge_sharded_state_dict support distribute merge (#75005)

* fix data is nullptr

* add dist merge

* change test

* change test

* 【FlexCP】add Skip param param for merge_shard_state_dict (#75061)

* fix data is nullptr

* add dist merge

* change test

* change test

* add skip optimizer param

* [Flex CP]Fix merge_sharded_state_dict with aoa and offload (#75062)

* fix merge_state_dict with aoa and offload

* add tests

* refine

* fix

* fix

* add log

* fix

* fix

* 【FlexCheckpoint】Upgrade some macros and optimize load_state_dict communication (#75282)

* upgrad macros and load_state_dict comm task

fix

fix

support 0-d tensor

fix

balance save and fix

* fix test

* Add the test about the sharded_state_dict of optimizer  (#75067)

* fix the share_weight_bug

* add note

* add the unit test

* set the timeout

* add more test

* Trigger CI rebuild

* fix the CmakeLists

* handle_missing_edge_cases_in_fc (#75413)

* up_grade fc (#75613)

fix and add test

fix

fix

fix

fix cmakelists

add notion

* 【FlexCheckpoint】fix_the_layer_id_macro (#75556)

* fix_the_layer_id_macro

* fix the ctest

* add expert_id_macro

* fix the assert bug

* fix the code style

* Pr support load hf checkpoint (#75928)

* support hf checkpoint

fix

support cast

add id macro

fix

* add test and fix some bug

* fix full param bug

* add full param cast test

---------

Co-authored-by: xingmingyyj <[email protected]>

* 【Flexcheckpoint】add_get_var_mapping_chain_macro (#76013)

* add_get_var_mapping_chain_macro

* add note

* fix the bug input_vars and resolve_mapping_chain

* fix the code style

* fit the dtype assert bug

* fix the bug

* fix the merge_sharded_state_dict bug

* fix aoa transpose corner case (#76234)

---------

Co-authored-by: xiaoguoguo626807 <[email protected]>
Co-authored-by: Chen Zhiyang <[email protected]>
Co-authored-by: Tianyu Zheng <[email protected]>
@zty-king zty-king deleted the handle_missing_edge_cases_in_fc branch November 23, 2025 13:55
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

5 participants