Add meta for _add_relu

pralay-das · pralay-das · commit 6765624b7333 · 2024-11-11T07:20:21.000+02:00
diff --git a/torch/_meta_registrations.py b/torch/_meta_registrations.py
@@ -3685,7 +3685,9 @@ def meta_relu_(self):
 @register_meta(aten._add_relu.Tensor)
 @out_wrapper()
 def meta__add_relu(self, other, alpha=1) -> Tensor:
-    return torch.empty_like(self)
+    return elementwise_meta(
+        self, other, type_promotion=ELEMENTWISE_TYPE_PROMOTION_KIND.DEFAULT
+    )
 
 
 @register_meta([aten.index_put.default, aten._unsafe_index_put.default])
