BUG Fixes regression for nllloss gradcheck (#64203)
authorThomas J. Fan <thomasjpfan@gmail.com>
Mon, 30 Aug 2021 22:03:40 +0000 (15:03 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Mon, 30 Aug 2021 22:13:09 +0000 (15:13 -0700)
commita7ae73a2380c3e45394998d2d1d9bceb14f2ee55
treeed2082605efe7942d224bcc9ac37a20f08d756b1
parentad4848565e1d9f4d408c60614f213acb52035181
BUG Fixes regression for nllloss gradcheck (#64203)

Summary:
Fixes https://github.com/pytorch/pytorch/issues/64163

This PR includes the fix and the opinfo from https://github.com/pytorch/pytorch/pull/63854/ for non-regression testing.

cc albanD mruberry jbschlosser

Pull Request resolved: https://github.com/pytorch/pytorch/pull/64203

Reviewed By: albanD

Differential Revision: D30647522

Pulled By: jbschlosser

fbshipit-source-id: 2974d299763505908fa93532aca2bd5d5b71f2e9
aten/src/ATen/native/cuda/Loss.cu
torch/testing/_internal/common_methods_invocations.py