Route matmul_kernel

src/fused_mm_sampling/persistent_matmul.py:None–None

triton.autotune(
    configs=matmul_get_configs(),
    key=["M", "N", "K"],
)

import argparse
import itertools
from contextlib import contextmanager
from typing import Optional