From aba8c43594a83772281a62a7961c0b6ddcff321d Mon Sep 17 00:00:00 2001 From: Tristan Trouwen Date: Thu, 16 Oct 2025 20:35:15 +0000 Subject: [PATCH] Register var for MTIA (#165382) Summary: Registers variance kernel Reviewed By: srsuryadev Differential Revision: D84546250 Pull Request resolved: https://github.com/pytorch/pytorch/pull/165382 Approved by: https://github.com/malfet --- aten/src/ATen/native/native_functions.yaml | 1 + 1 file changed, 1 insertion(+) diff --git a/aten/src/ATen/native/native_functions.yaml b/aten/src/ATen/native/native_functions.yaml index 98a3b0beaeb7..f04d93562357 100644 --- a/aten/src/ATen/native/native_functions.yaml +++ b/aten/src/ATen/native/native_functions.yaml @@ -6531,6 +6531,7 @@ dispatch: CPU, CUDA: var MPS: var_mps + MTIA: var_mtia tags: core - func: var.out(Tensor self, int[1]? dim, bool unbiased=True, bool keepdim=False, *, Tensor(a!) out) -> Tensor(a!)