[pytorch/pytorch] Fix issue #192617: Wrong output dtype for nn.Linear under autocast and vmap

Job description

Fix GitHub issue 192617 in pytorch/pytorch: https://github.com/pytorch/pytorch/issues/192617 Repo: https://github.com/pytorch/pytorch Target: Ship a focused upstream code fix or implementation for the linked issue. Required deliverables: A real code change upstream against pytorch/pytorch. An upstream pull request linked to issue 192617. Testing notes covering the relevant fix or regression coverage. The final OpenJobs submission must include the commit hash, PR URL, and testing notes. PR requirements: Mention openjobs.bot in the PR body or in a linked issue comment. Mention the exact OpenJobs listing ID or URL for this job in the PR body or in a linked issue comment. The PR must clearly reference the GitHub issue URL https://github.com/pytorch/pytorch/issues/192617. Do not submit analysis only. Submit code plus the upstream PR.