pytorch

5112f44d - Add vmap support for torch.index_fill (#91364)

Add vmap support for torch.index_fill (#91364) Fixes #91177 Pull Request resolved: https://github.com/pytorch/pytorch/pull/91364 Approved by: https://github.com/zou3519