xma.inductor

enable_kernels(kernels: list[str], _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) → 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]]]

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