Tags: NodeGraphSystems/pytorch
Tags
Use counter instead of vector of futures in `_parallel_run` (pytorch#… …36159) (pytorch#36334) Summary: This should be faster than allocating one mutex, flag and conditional variable per task. Using `std::atomic<size_t>` to count remaing tasks is not sufficient, because modification of remaining counter and signalling conditional variable must happen atomically, otherwise `wait()` might get invoked after `notify_one()` was called. Pull Request resolved: pytorch#36159 Test Plan: CI Differential Revision: D20905411 Pulled By: malfet fbshipit-source-id: facaf599693649c3f43edafc49f369e90d2f60de (cherry picked from commit 986a8fd) Signed-off-by: Eli Uriegas <[email protected]> Co-authored-by: Nikita Shulga <[email protected]>
Revert "Fix handling of non-finite values in topk (pytorch#35253)" (p… …ytorch#35582) This reverts commit b12579d. This patch in-and-of itself looks fine, but it's causing some AMP tests to fail.
.circleci: Remove quotes from --git-dir git doesn't handle the escapes correctly so let's just not put them altogether. Signed-off-by: Eli Uriegas <[email protected]>
Enabling inplace relu Summary: Pull Request resolved: pytorch#28710 Test Plan: Imported from OSS Differential Revision: D18146120 Pulled By: z-a-f fbshipit-source-id: d8f0982f5a2ae35f7deb34e67cdb64be700a9d6c
Add autofunctions in torch.rst This is the v1.3.0 version of a 3 Part PR originally made to master PR: pytorch#27677 Originally by @dzhulgakov
Roll master to 1.4.0 (pytorch#27374) Summary: Pull Request resolved: pytorch#27374 Signed-off-by: Edward Z. Yang <[email protected]> Test Plan: Imported from OSS Differential Revision: D17809770 Pulled By: ezyang fbshipit-source-id: 75bd97426494a7bbbf08f9bce7563d35871443d8
Roll master to 1.3.0 (pytorch#23895) Summary: Pull Request resolved: pytorch#23895 Signed-off-by: Edward Z. Yang <[email protected]> Test Plan: Imported from OSS Differential Revision: D16688489 Pulled By: ezyang fbshipit-source-id: a56d0180a0bc57775badd9e31ea3d441d5fd4f88
PreviousNext