Diff Coverage

Total: 55 lines
Missing: 4 lines
Coverage: 92%

Source File	Diff Coverage (%)	Missing Lines
hyper_parallel/core/pipeline_parallel/scheduler.py	92.7%	2522,2547,2597-2598

hyper_parallel/core/pipeline_parallel/scheduler.py

    stage_order = []
    for action_index, action in enumerate(actions):
        for leaf_step in iter_leaf_meta_steps(action):
            if leaf_step.stage_index not in managed_stage_indices:
                continue
            if leaf_step.micro_index != last_micro:
                continue
            is_backward_leaf = leaf_step.type in (
                MetaStepType.BWD,

                reduce_points[stage_index] = max(
                    reduce_points[stage_index], action_index + 1
                )
                if is_backward_leaf:
                    launch_points[stage_index] = max(
                        launch_points[stage_index], action_index + 1
                    )
    return stage_order, reduce_points, launch_points

                actions[index], stage_index, MetaStepType.FSDP_UNSHARD
            )
            for index in window
        ):
            reduce_points[stage_index] = launch_point
            continue
        if reshard_indices:
            relocated_reshards[stage_index] = reshard_indices
            skipped_action_indices.update(reshard_indices)
    return relocated_reshards, skipped_action_indices