(precision, profile_name)
| 818 | |
| 819 | |
| 820 | def show_profile(precision, profile_name): |
| 821 | import triton.profiler.viewer as proton_viewer |
| 822 | |
| 823 | metric_names = ["time/ms"] |
| 824 | if precision == "fp8": |
| 825 | metric_names = ["tflop8/s"] + metric_names |
| 826 | elif precision == "fp16": |
| 827 | metric_names = ["tflop16/s"] + metric_names |
| 828 | file_name = f"{profile_name}.hatchet" |
| 829 | tree, metrics = proton_viewer.parse(metric_names, file_name) |
| 830 | proton_viewer.print_tree(tree, metrics) |
| 831 | |
| 832 | |
| 833 | def main(): |