pytorch
7b0e140e - [1.11] Remove torch.vmap (#65496) (#72275)

Commit
3 years ago
[1.11] Remove torch.vmap (#65496) (#72275) * release 1.11 Install torch from test channel, Pin builder and xla repo (#72217) * [1.11] Remove torch.vmap (#65496) torch.vmap is a prototype feature and should not be in the stable binary. This PR: - Removes the torch.vmap API - Removes the documentation entry for torch.vmap - Changes the vmap tests to use an internal API instead of torch.vmap. Test Plan: - Tested locally (test_torch, test_autograd, test_type_hints, test_vmap), but also wait for CI. Co-authored-by: Andrey Talman <atalman@fb.com>
Author
Parents
Loading