add type promotion wrapper

rdspring1 · rdspring1 · commit 3cd82ab2f51a · 2022-10-16T18:35:02.000-07:00
diff --git a/torch/_refs/nn/functional/__init__.py b/torch/_refs/nn/functional/__init__.py
@@ -516,6 +516,10 @@ def _nll_loss_nd(
 
 
 @register_decomposition(torch.ops.aten.nll_loss)
+@elementwise_type_promotion_wrapper(
+    type_promoting_args=("input",),
+    type_promotion_kind=ELEMENTWISE_TYPE_PROMOTION_KIND.DEFAULT,
+)
 @out_wrapper()
 def nll_loss(
     input: TensorLikeType,
