xma.inductor

enable_kernels(kernels: list[tuple[str, KernelBackend | None]], _patterns: torch._inductor.pattern_matcher.PatternMatcherPass = torch._inductor.fx_passes.joint_graph.patterns, device: device = None) → None

get_fused_residual_add_rmsnorm_replacer(device: torch.device, kernel_backend: KernelBackend | None = None) → Generator[tuple[Callable, Callable, tuple[torch.Tensor, torch.Tensor, torch.Tensor]]]

get_rmsnorm_replacer(device: torch.device, kernel_backend: KernelBackend | None = None) → Generator[tuple[Callable, Callable, tuple[torch.Tensor, torch.Tensor]]]

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