wafer-core 0.1.33__py3-none-any.whl → 0.1.34__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- wafer_core/lib/trace_compare/__init__.py +9 -22
- wafer_core/lib/trace_compare/analyzer.py +160 -584
- wafer_core/lib/trace_compare/classifier.py +18 -321
- wafer_core/lib/trace_compare/fusion_analyzer.py +753 -329
- wafer_core/lib/trace_compare/loader.py +220 -413
- wafer_core/targets/__init__.py +21 -47
- wafer_core/utils/kernel_utils/defense.py +1 -813
- wafer_core/utils/kernel_utils/targets/config.py +24 -8
- {wafer_core-0.1.33.dist-info → wafer_core-0.1.34.dist-info}/METADATA +1 -1
- {wafer_core-0.1.33.dist-info → wafer_core-0.1.34.dist-info}/RECORD +11 -11
- {wafer_core-0.1.33.dist-info → wafer_core-0.1.34.dist-info}/WHEEL +0 -0
|
@@ -5,41 +5,28 @@ identifying kernel-level performance differences and fusion opportunities.
|
|
|
5
5
|
"""
|
|
6
6
|
|
|
7
7
|
from .analyzer import analyze_traces
|
|
8
|
-
from .api import TraceComparisonResult, analyze_trace_pair
|
|
9
|
-
from .architecture import ArchitectureType, detect_architecture
|
|
10
8
|
from .classifier import Op, classify
|
|
11
|
-
from .aligner import align_traces, TraceAlignment, LayerAlignment, KernelPair
|
|
12
9
|
from .formatter import (
|
|
13
10
|
format_csv,
|
|
11
|
+
format_fusion_csv,
|
|
12
|
+
format_fusion_json,
|
|
13
|
+
format_fusion_text,
|
|
14
14
|
format_json,
|
|
15
15
|
format_text,
|
|
16
16
|
)
|
|
17
|
-
from .fusion_analyzer import
|
|
18
|
-
from .
|
|
19
|
-
from .loader import load_trace, load_trace_streaming, StreamingMetadata
|
|
20
|
-
from .warnings import TraceWarning, detect_warnings
|
|
17
|
+
from .fusion_analyzer import analyze_fusion_differences
|
|
18
|
+
from .loader import load_trace
|
|
21
19
|
|
|
22
20
|
__all__ = [
|
|
23
21
|
"Op",
|
|
24
22
|
"classify",
|
|
25
23
|
"load_trace",
|
|
26
|
-
"load_trace_streaming",
|
|
27
|
-
"StreamingMetadata",
|
|
28
24
|
"analyze_traces",
|
|
25
|
+
"analyze_fusion_differences",
|
|
29
26
|
"format_text",
|
|
30
27
|
"format_csv",
|
|
31
28
|
"format_json",
|
|
32
|
-
|
|
33
|
-
"
|
|
34
|
-
"
|
|
35
|
-
"ArchitectureType",
|
|
36
|
-
"detect_architecture",
|
|
37
|
-
"TraceWarning",
|
|
38
|
-
"detect_warnings",
|
|
39
|
-
"align_traces",
|
|
40
|
-
"TraceAlignment",
|
|
41
|
-
"LayerAlignment",
|
|
42
|
-
"KernelPair",
|
|
43
|
-
"analyze_fusion_from_alignment",
|
|
44
|
-
"analyze_same_kernels_from_alignment",
|
|
29
|
+
"format_fusion_text",
|
|
30
|
+
"format_fusion_csv",
|
|
31
|
+
"format_fusion_json",
|
|
45
32
|
]
|