xma.inductor

enable_kernels(kernels: list[str])

get_fused_residual_add_rmsnorm_replacer(device: torch.device) → Generator[tuple[Callable, Callable, tuple[torch.Tensor, torch.Tensor, torch.Tensor]]]

get_rmsnorm_replacer(device: torch.device) → Generator[tuple[Callable, Callable, tuple[torch.Tensor, torch.Tensor]]]

init_inductor(cache_size_limit: int) → None

partialize_and_update_signature(func: Callable, **kwargs) → Callable