Masaki Kozuki
Masaki Kozuki
> The implementation of this monster of a function is also located at `NLLLoss2d.cu`. Could you apply similar changes to that branch, and add the relevant tests to `test_nn.py`? From...
then I think it's already covered as `nll_loss_nd` is what `torch.nn.functional.nll_loss` calls: https://github.com/pytorch/pytorch/blob/291b080e8c54cdb19dd823728cf30fffece10f4d/torch/nn/functional.py#L2697
@pytorchbot merge
@ngimel or @albanD could you take a look at this PR? fwiw, this isn't dependent on #81705
will rerun the CI... as the failure is unrelated ``` a231f0614b3d: Download complete 7f159b2249ec: Verifying Checksum 7f159b2249ec: Download complete write /var/lib/docker/tmp/GetImageBlob53272514: no space left on device Error: Process completed with...
@pytorchmergebot rebase
@pytorchmergebot rebase
@pytorchbot rebase
In https://github.com/pytorch/pytorch/actions/runs/3156151643/jobs/5135717985 ``` Traceback (most recent call last): File "/var/lib/jenkins/workspace/test/run_test.py", line 1161, in main() File "/var/lib/jenkins/workspace/test/run_test.py", line 1139, in main raise RuntimeError(err_message) RuntimeError: test_ops failed! ``` @pytorchbot merge -f "test_ops.py...
cc @ChongyuNVIDIA @jpool-nv