PR #41860 [tests] Add Context-parallel CI tests

[tests] Add Context-parallel CI tests #41860

SunMarc merged 12 commits into main from cp-ci-tests

intial

63efa183

simplify tests

1ff587b5

add test_cp_equivalence

0bcf34ed

kashif requested a review from

SunMarc 57 days ago

removed fsdp_transformer_layer_cls_to_wrap

6d42d9a5

use DataCollatorForLanguageModeling

e18436c0

remove use_cache=False.

df8aaacb

kashif requested a review from

ydshieh 57 days ago

stas00 commented on 2025-10-26

ydshieh commented on 2025-10-27

changes from review

4ec11c20

make script self contained

a4a187e0

kashif requested a review from

ydshieh 48 days ago

Merge branch 'main' into cp-ci-tests

977c5865

SunMarc approved these changes on 2025-11-04

moved to fsdp folder

5a7d1aa7

fix class name

8b39cd53

Merge branch 'main' into cp-ci-tests

9bd5dae6

SunMarc merged 0c4a2024 into main 46 days ago

SunMarc deleted the cp-ci-tests branch 46 days ago

Reviewers

SunMarc

ydshieh

stas00

Assignees

No one assigned

Labels

None yet

Milestone

No milestone

transformers [tests] Add Context-parallel CI tests #41860 Merged

[tests] Add Context-parallel CI tests #41860

transformers
[tests] Add Context-parallel CI tests
#41860

Merged