tilelang.testing.perf_regression

Attributes

Classes

Functions

tqdm(iterable, **kwargs)

process_func(func[, name])

Execute a single perf function and record its latency.

regression([prefixes])

Run entrypoints in the caller module and print a markdown table.

regression_all([examples_root])

Run all example benchmark drivers and print a consolidated table.

Module Contents

tilelang.testing.perf_regression.tabulate = None
tilelang.testing.perf_regression.tqdm(iterable, **kwargs)
class tilelang.testing.perf_regression.PerfResult
name: str
latency: float
tilelang.testing.perf_regression.process_func(func, name=None, /, **kwargs)

Execute a single perf function and record its latency.

func is expected to return a positive latency scalar (seconds or ms; we treat it as an opaque number, only ratios matter for regression).

Parameters:
  • func (Callable[Ellipsis, float])

  • name (str | None)

  • kwargs (Any)

Return type:

float

tilelang.testing.perf_regression.regression(prefixes=('regression_',))

Run entrypoints in the caller module and print a markdown table.

This is invoked by many example scripts.

Parameters:

prefixes (collections.abc.Sequence[str])

Return type:

None

tilelang.testing.perf_regression.regression_all(examples_root=None)

Run all example benchmark drivers and print a consolidated table.

Intended usage (CI): python -c “import tilelang.testing.perf_regression as pr; pr.regression_all()”

Parameters:

examples_root (str | os.PathLike[str] | None)

Return type:

None