temporary xfail for nll_loss

Author: zou3519

test/test_ops.py

@@ -351,6 +351,7 @@ def vjp_of_vjp(*args_and_cotangents):
         xfail('block_diag'),
         xfail('nn.functional.dropout'),
         xfail('nn.functional.max_pool2d'),
+        xfail('nn.functional.nll_loss'),
     }))
     def test_vmapvjp(self, device, dtype, op):
         # These are too annoying to put into the list above
@@ -525,6 +526,7 @@ def test():
         xfail('block_diag'),
         xfail('nn.functional.max_pool2d'),
         xfail('nn.functional.batch_norm'),
+        xfail('nn.functional.nll_loss'),
     }))
     def test_vjpvmap(self, device, dtype, op):
         # NB: there is no vjpvmap_has_batch_rule test because that is almost

test/test_vmap.py

@@ -3001,6 +3001,7 @@ class TestVmapOperatorsOpInfo(TestCase):
         xfail('index_put'),
         xfail('nn.functional.max_pool2d'),
         xfail('nn.functional.batch_norm'),
+        xfail('nn.functional.nll_loss'),
     })
     def test_vmap_exhaustive(self, device, dtype, op):
         sample_inputs_itr = op.sample_inputs(device, dtype, requires_grad=False)