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 the four scattermoe-lora kernels.

collect_autotune_configs

integrations.kernels.autotune_collector.collect_autotune_configs()

Read autotune caches from the four scattermoe-lora kernels.

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

• kernel – human-readable kernel name
• key – dict with the M/N/K problem dimensions
• config – dict with the selected tile sizes, num_warps, and num_stages

Returns [] if the kernel module cannot be found or if no autotune cache entries exist yet.