integrations.kernels.autotune_collector

integrations.kernels.autotune_collector

Collect Triton autotune results from scattermoe-lora kernels.

This module reads the .cache attribute from Triton @triton.autotune decorated kernel objects and returns structured dicts describing the selected configurations. It has no telemetry dependency — callers decide what to do with the data.

Functions

Name Description
collect_autotune_configs Read autotune caches from ALL dsv4 + scattermoe @triton.autotune kernels.

collect_autotune_configs

integrations.kernels.autotune_collector.collect_autotune_configs()

Read autotune caches from ALL dsv4 + scattermoe @triton.autotune kernels.

Returns a (possibly empty) list of dicts, each containing:

  • kernel – kernel function name
  • module – short module name it lives in
  • key – dict of the autotune-key args (problem shapes), labeled by the kernel’s own declared key names
  • config – selected tile sizes, num_warps, num_stages (and num_ctas)

Scans sys.modules for both the HF-kernels hash-suffixed scattermoe copies (whose caches are the populated runtime ones) and the normally-imported dsv4 kernel modules.