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

1# Copyright 2026 Huawei Technologies Co., Ltd 

2# Licensed under the Apache License, Version 2.0 

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

4"""Codegen: bake TP/CP/EP parallel contracts into a generated HF-modeling file. 

5 

6Public manager helpers: 

7 

8- :func:`ensure_codegen_artifact` — generate/reuse the bundle for a YAML. 

9- :func:`preflight_integrity_check` — fail-fast drift detection before train. 

10- :func:`resolve_artifact_layout` — map a YAML path to its bundle paths. 

11 

12Planning and validation APIs: 

13 

14- :func:`derive_sharding_plan` / :func:`freeze_plan` — run the planner over 

15 an offline mesh and freeze the plan into JSON-safe structures. 

16- :func:`verify_signature` / :func:`verify_output_hashes` / ... — per-zone 

17 preflight verifiers. 

18""" 

19from hyper_parallel.codegen.artifact import resolve_artifact_layout 

20from hyper_parallel.codegen.check.preflight import ( 

21 verify_generated_import, 

22 verify_meta_required_fields, 

23 verify_output_hashes, 

24 verify_param_plan, 

25 verify_signature, 

26 warn_skipped_overrides, 

27) 

28from hyper_parallel.codegen.manager import ( 

29 ensure_codegen_artifact, 

30 preflight_integrity_check, 

31) 

32from hyper_parallel.codegen.plan.freeze import ( 

33 FrozenPlan, 

34 freeze_plan, 

35 freeze_param_plan, 

36 named_placement_to_dict, 

37 placement_to_string, 

38) 

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

40 

41__all__ = [ 

42 "FrozenPlan", 

43 "OfflineMesh", 

44 "build_offline_mesh", 

45 "ensure_codegen_artifact", 

46 "freeze_param_plan", 

47 "freeze_plan", 

48 "named_placement_to_dict", 

49 "placement_to_string", 

50 "preflight_integrity_check", 

51 "resolve_artifact_layout", 

52 "verify_generated_import", 

53 "verify_meta_required_fields", 

54 "verify_output_hashes", 

55 "verify_param_plan", 

56 "verify_signature", 

57 "warn_skipped_overrides", 

58]