From ea642a3fd5e31a2778aa588d344529829df93cf8 Mon Sep 17 00:00:00 2001
From: sp4s-s <168576652+sp4s-s@users.noreply.github.com>
Date: Mon, 14 Sep 2026 13:56:31 +0530
Subject: [PATCH] restoring perf regression for upsample `linear` mode
 align_corners=False

Results, Accuracy [at]
https://github.com/sp4s-s/mlx/blob/development/results.md
---
 python/mlx/nn/layers/upsample.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/python/mlx/nn/layers/upsample.py b/python/mlx/nn/layers/upsample.py
index 0be9f87bfb..b04f6e9ba7 100644
--- a/python/mlx/nn/layers/upsample.py
+++ b/python/mlx/nn/layers/upsample.py
@@ -302,6 +302,13 @@ def upsample_linear(
             indices_fn=_linear_aa_indices,
             align_corners=align_corners,
         )
+    if not antialias and not align_corners:
+        return _interpolate_separable(
+            x=x,
+            scale_factor=scale_factor,
+            indices_fn=_linear_indices,
+            align_corners=align_corners,
+        )
     return _interpolate(
         x=x,
         scale_factor=scale_factor,