frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Revert "[Compiled Autograd] Move to torch::dynamo::autograd namespace (#105854)"

Browse Source 
This reverts commit 26e3b4020f01d4fc2b7f63e1de4c94d2c8b362b5.

Reverted https://github.com/pytorch/pytorch/pull/105854 on behalf of https://github.com/PaliC due to breaking internal embedded device tests (details shared with author) ([comment](https://github.com/pytorch/pytorch/pull/105854#issuecomment-1650559375))
This commit is contained in:
PyTorch MergeBot
2023-07-25 21:09:18 +00:00
parent a4cffaae67
commit e60af5c8e4
8 changed files with 23 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
torch/csrc/autograd/function.h
View File

@ -40,14 +40,14 @@ namespace autograd {
struct Edge;
struct FunctionPostHook;
struct FunctionPreHook;
class CompiledNodeArgs;
class SwapSavedVariables;
using tensor_list = std::vector<at::Tensor>;
using variable_list = std::vector<Variable>;
using edge_list = std::vector<Edge>;
using saved_variable_list = std::vector<SavedVariable>;
using IndexRange = std::pair<size_t, size_t>;
using torch::dynamo::autograd::CompiledNodeArgs;
using torch::dynamo::autograd::SwapSavedVariables;
// Custom deleter to prevent stack overflows.
TORCH_API void deleteNode(Node* function);
@ -578,7 +578,7 @@ struct TORCH_API Node : std::enable_shared_from_this<Node> {
// Used by compiled autograd to call apply() with different saved tensors
// Implementations should call saved.before() on all attrs, then apply(), then
// saved.after() on all attrs in the same order.
// saved.after() on all attrs.
virtual variable_list apply_with_saved(
const variable_list& inputs,
SwapSavedVariables& saved) {