Coverage for  / home / jenkins / .local / lib / python3.10 / site-packages / hyper_parallel / codegen / plan / __init__.py: 100%

1# Copyright 2026 Huawei Technologies Co., Ltd 

2# Licensed under the Apache License, Version 2.0 

3# ============================================================================ 

4"""Plan derivation: offline mesh + ShardingPlanner -> frozen plan. 

5 

6The planner runs at generation time over a static offline mesh and 

7freezes the result into ``CodegenMeta`` so the generated modeling file (and 

8preflight) can consume it without re-running the planner. 

9""" 

10from hyper_parallel.codegen.plan.boundary_forms import ( 

11 BoundaryForm, 

12 FORM_GENERIC, 

13 FORM_IDENTITY, 

14 FORM_REGION, 

15 FORM_TP_COLLECTIVE, 

16 TransitionOp, 

17 classify_boundary_form, 

18) 

19from hyper_parallel.codegen.plan.derive import ( 

20 build_meta_model, 

21 build_plan_overrides, 

22 derive_sharding_plan, 

23) 

24from hyper_parallel.codegen.plan.freeze import ( 

25 FrozenPlan, 

26 expand_frozen_sharded_params, 

27 freeze_injections, 

28 freeze_param_plan, 

29 freeze_plan, 

30 freeze_tied_pairs, 

31 named_placement_to_dict, 

32 placement_to_string, 

33) 

34from hyper_parallel.codegen.plan.offline_mesh import OfflineMesh, build_offline_mesh 

35 

36__all__ = [ 

37 "BoundaryForm", 

38 "FORM_GENERIC", 

39 "FORM_IDENTITY", 

40 "FORM_REGION", 

41 "FORM_TP_COLLECTIVE", 

42 "FrozenPlan", 

43 "OfflineMesh", 

44 "TransitionOp", 

45 "build_meta_model", 

46 "build_offline_mesh", 

47 "build_plan_overrides", 

48 "classify_boundary_form", 

49 "derive_sharding_plan", 

50 "expand_frozen_sharded_params", 

51 "freeze_injections", 

52 "freeze_param_plan", 

53 "freeze_plan", 

54 "freeze_tied_pairs", 

55 "named_placement_to_dict", 

56 "placement_to_string", 

57]