diff --git a/aten/src/ATen/native/native_functions.yaml b/aten/src/ATen/native/native_functions.yaml index ef734f875deb..3ed0592b764d 100644 --- a/aten/src/ATen/native/native_functions.yaml +++ b/aten/src/ATen/native/native_functions.yaml @@ -5345,7 +5345,7 @@ - func: logit(Tensor self, float? eps=None) -> Tensor variants: function, method dispatch: - CPU, CUDA: logit + CPU, CUDA, MTIA: logit MPS: logit_mps tags: pointwise