diff --git a/src/triton_cli/profiler.py b/src/triton_cli/profiler.py index 951ebbc..48a489f 100644 --- a/src/triton_cli/profiler.py +++ b/src/triton_cli/profiler.py @@ -27,7 +27,7 @@ import json import subprocess from dataclasses import dataclass -from itertools import pairwise +from itertools import tee from pathlib import Path from typing import Optional @@ -75,6 +75,14 @@ } +# Built-in to itertools in Python 3.10+ +def pairwise(iterable): + # n=2 for pairs + a, b = tee(iterable, 2) + next(b, None) + return zip(a, b) + + @dataclass class ProfileResults: prompt_size: int