Consolidate with div

pytorch · Jul 13, 2023 · 7b41a20 · 7b41a20
1 parent 08a3bad
commit 7b41a20
Showing 1 changed file with 1 addition and 6 deletions.
diff --git a/torch/_meta_registrations.py b/torch/_meta_registrations.py
@@ -2733,12 +2733,7 @@ def meta__foreach_add__list(self, other, alpha=1):
     _check_foreach_binop_tensor_lists(self, other)
 
 
-@register_meta([aten._foreach_mul_.List])
-def meta__foreach_mul__list(self, other):
-    _check_foreach_binop_tensor_lists(self, other)
-
-
-@register_meta([aten._foreach_div_.List])
+@register_meta([aten._foreach_mul_.List, aten._foreach_div_.List])
 def meta__foreach_binop__list(self, other):
     _check_foreach_binop_tensor_lists(self, other)