Skip to content

[fix] Enforce param dtype before wrap ddp - #24

Merged
Zhichenzzz merged 1 commit into
miles-mainfrom
fix/fp32_mark
Apr 20, 2026
Merged

[fix] Enforce param dtype before wrap ddp#24
Zhichenzzz merged 1 commit into
miles-mainfrom
fix/fp32_mark

Conversation

@guapisolo

@guapisolo guapisolo commented Apr 16, 2026

Copy link
Copy Markdown

No description provided.

@guapisolo guapisolo changed the title [fix] fp32 mark [fix] get model sliently changed to fp32 Apr 16, 2026
@guapisolo guapisolo changed the title [fix] get model sliently changed to fp32 [fix] Enforce param dtype before wrap ddp Apr 16, 2026
@Zhichenzzz
Zhichenzzz merged commit d9a5080 into miles-main Apr 20, 2026
fzyzcjy added a commit that referenced this pull request Jun 5, 2026
…ixes)

Brings in: true-on-policy init (mappings.py _tree_all_reduce_sum +
_DeterministicReduceFromModelParallelRegion, miles_megatron_plugins/true_on_policy,
gpt_layer_specs use_true_on_policy_backend) + plugin packaging (#49, #50) + fixes
(param dtype before DDP wrap #24, attention_output_gate TP slice #22, fp32 shard
detach).
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants