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

1# Copyright 2026 Huawei Technologies Co., Ltd 

2# 

3# Licensed under the Apache License, Version 2.0 (the "License"); 

4# you may not use this file except in compliance with the License. 

5# You may obtain a copy of the License at 

6# 

7# http://www.apache.org/licenses/LICENSE-2.0 

8# 

9# Unless required by applicable law or agreed to in writing, software 

10# distributed under the License is distributed on an "AS IS" BASIS, 

11# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 

12# See the License for the specific language governing permissions and 

13# limitations under the License. 

14# ============================================================================ 

15"""In-memory inputs consumed while emitting one generated bundle.""" 

16 

17from __future__ import annotations 

18 

19from dataclasses import dataclass, field 

20from typing import Any, Optional 

21 

22 

23@dataclass 

24class EmissionContext: 

25 """Hold frozen native-build facts for a single emission pass.""" 

26 

27 source: dict[str, Any] 

28 parallel_dims: dict[str, Any] 

29 status: str 

30 injections: list[dict[str, Any]] = field(default_factory=list) 

31 param_plan: dict[str, Any] = field(default_factory=dict) 

32 boundary_classes: dict[str, str] = field(default_factory=dict) 

33 nested_components: list[dict[str, str]] = field(default_factory=list) 

34 module_overrides: list[dict[str, Any]] = field(default_factory=list) 

35 materialized_patches: list[dict[str, str]] = field(default_factory=list) 

36 emission_patches: dict[str, Any] = field(default_factory=dict) 

37 external_state_classes: list[str] = field(default_factory=list) 

38 mesh_dim_names: Optional[list[str]] = None 

39 

40 

41__all__ = ["EmissionContext"]