[triton] Tutorials: 벤치마크 결과 테이블에 단위(units) 표시 추가

PR 링크: triton-lang/triton#8631 상태: Merged | 변경: +8 / -8

들어가며

Triton의 벤치마크 유틸리티는 성능 측정 결과를 테이블과 그래프로 출력합니다. 기존에는 컬럼명에 단위(예: GB/s, TFLOPS)가 포함되지 않아, 수치만으로는 결과의 의미를 파악하기 어려웠습니다. 이번 PR은 ylabel을 컬럼명에 포함시켜 이 문제를 해결합니다.

핵심 코드 분석

Before: 단위 없는 컬럼명

# testing.py
y_mean = bench.line_names           # ['Triton', 'Torch']
y_min = [f'{x}-min' for x in bench.line_names]
y_max = [f'{x}-max' for x in bench.line_names]

출력 예시:

size  Triton  Torch
4096  8.83    8.78

After: 단위 포함 컬럼명

# testing.py
y_mean_labels = [f'{x} ({bench.ylabel})' for x in bench.line_names]
y_min_labels = [f'{x}-min ({bench.ylabel})' for x in bench.line_names]
y_max_labels = [f'{x}-max ({bench.ylabel})' for x in bench.line_names]

출력 예시:

size  Triton (GB/s)  Torch (GB/s)
4096  8.83           8.78

왜 이게 좋은가

1. 즉시 이해 가능: 숫자가 GB/s인지, TFLOPS인지, ms인지 테이블만 보고 바로 알 수 있습니다.
2. 최소 변경: 8줄만 바꿔서 모든 튜토리얼의 벤치마크 출력을 개선합니다.
3. 그래프 레이블도 일관: 그래프의 범례(legend)에도 단위가 표시됩니다.

정리

작은 UX 개선이지만, 튜토리얼이 참고 문서로 활용되는 점을 고려하면 중요한 변경입니다. 벤치마크 결과의 단위를 명시하는 것은 과학적 의사소통의 기본 원칙이기도 합니다.

참고 자료

• triton-lang/triton#8631
• GitHub Issue #8588

---

이 글은 AI(Claude)의 도움을 받아 작성되었으며, PR의 실제 diff를 기반으로 분석한 내용입니다.