| # Copyright(C) [2025] Advanced Micro Devices, Inc. All rights reserved. | |
| import os | |
| from .perf.efficiency import PerformanceEvalTBG, PerformanceEvalROCm | |
| from .constants import TBG_PERF_GOLD_ROOT, TBG_DATA_ROOT, ROCm_DATA_ROOT, NATIVE_PERF_GOLD_ROOT, ROCM_PERF_GOLD_DATA_ROOT | |
| def initialize_performance_eval_tb(): | |
| perf_evaluator = PerformanceEvalTBG() | |
| perf_evaluator.ref_folder = NATIVE_PERF_GOLD_ROOT | |
| print(f"Creating performance evaluation folder at {TBG_PERF_GOLD_ROOT}") | |
| perf_evaluator(exec_folder=TBG_DATA_ROOT, gen_perf_folder=TBG_PERF_GOLD_ROOT, golden_metrics_folder=NATIVE_PERF_GOLD_ROOT) | |
| def initialize_performance_eval_rocm(): | |
| perf_evaluator = PerformanceEvalROCm() | |
| perf_evaluator.ref_folder = ROCM_PERF_GOLD_DATA_ROOT | |
| # print(f"Creating performance evaluation folder at {TBG_PERF_GOLD_ROOT}") | |
| perf_evaluator(exec_folder=ROCm_DATA_ROOT, gen_perf_folder=TBG_PERF_GOLD_ROOT, golden_metrics_folder=NATIVE_PERF_GOLD_ROOT) | |
| if __name__ == "__main__": | |
| initialize_performance_eval_tb() | |