Revert "[MPS][BE] Delete unused lerp functors (#152443)"

pytorchmergebot · pytorchmergebot · commit e35e31697eb8 · 2025-04-29T22:50:23.000Z
This reverts commit 0a2d320. Reverted #152443 on behalf of https://github.com/wdvr due to failing MPS test: test/test_optim.py::TestOptimRenewedMPS::test_can_load_from_to_named_state_dict_is_named_optim0_False_is_named_optim1_False_Adafactor_mps_float32 ([comment](#152443 (comment)))
diff --git a/aten/src/ATen/native/mps/kernels/BinaryKernel.metal b/aten/src/ATen/native/mps/kernels/BinaryKernel.metal
@@ -18,6 +18,13 @@ struct sub_functor {
   }
 };
 
+struct lerp_functor {
+  template <typename T>
+  inline T operator()(const T a, const T b) {
+    return static_cast<T>(b);
+  }
+};
+
 struct add_alpha_functor {
   template <typename T>
   inline T operator()(const T a, const T b, const T alpha) {
@@ -222,6 +229,13 @@ struct complex_lerp_alpha_functor {
   }
 };
 
+struct complex_lerp_functor {
+  template <typename T>
+  inline T operator()(const T a, const T b) {
+    return T(b.x, b.y);
+  }
+};
+
 REGISTER_BINARY_OP(copysign, long, float);
 REGISTER_BINARY_OP(copysign, int, float);
 REGISTER_BINARY_OP(copysign, float, float);
@@ -268,6 +282,14 @@ REGISTER_BINARY_OP(sub, short, short);
 REGISTER_BINARY_OP(sub, uchar, uchar);
 REGISTER_BINARY_OP(sub, char, char);
 REGISTER_BINARY_OP(sub, bool, bool);
+REGISTER_BINARY_OP(lerp, long, long);
+REGISTER_BINARY_OP(lerp, int, int);
+REGISTER_BINARY_OP(lerp, float, float);
+REGISTER_BINARY_OP(lerp, half, half);
+REGISTER_BINARY_OP(lerp, short, short);
+REGISTER_BINARY_OP(lerp, uchar, uchar);
+REGISTER_BINARY_OP(lerp, char, char);
+REGISTER_BINARY_OP(lerp, bool, bool);
 REGISTER_BINARY_ALPHA_OP(add_alpha, long, long);
 REGISTER_BINARY_ALPHA_OP(add_alpha, int, int);
 REGISTER_BINARY_ALPHA_OP(add_alpha, float, float);
@@ -308,6 +330,7 @@ REGISTER_BINARY_OP(hermite_polynomial_h, bfloat, bfloat);
 REGISTER_BINARY_OP(hermite_polynomial_he, bfloat, bfloat);
 REGISTER_BINARY_OP(add, bfloat, bfloat);
 REGISTER_BINARY_OP(sub, bfloat, bfloat);
+REGISTER_BINARY_OP(lerp, bfloat, bfloat);
 REGISTER_BINARY_ALPHA_OP(add_alpha, bfloat, bfloat);
 REGISTER_BINARY_ALPHA_OP(sub_alpha, bfloat, bfloat);
 REGISTER_BINARY_ALPHA_OP(lerp_alpha, bfloat, bfloat);
@@ -324,6 +347,8 @@ REGISTER_BINARY_OP(add, float2, float2);
 REGISTER_BINARY_OP(add, half2, half2);
 REGISTER_BINARY_OP(sub, float2, float2);
 REGISTER_BINARY_OP(sub, half2, half2);
+REGISTER_BINARY_OP(lerp, float2, float2);
+REGISTER_BINARY_OP(lerp, half2, half2);
 REGISTER_BINARY_ALPHA_OP(complex_add_alpha, float2, float2);
 REGISTER_BINARY_ALPHA_OP(complex_add_alpha, half2, half2);
 REGISTER_BINARY_ALPHA_OP(complex_sub_alpha, float2, float2);
