From 47e27723203e40074571fa58343dd7cfe3e88ab6 Mon Sep 17 00:00:00 2001 From: Pieter Noordhuis Date: Fri, 29 Mar 2019 16:15:10 -0700 Subject: [PATCH] Update argument names of torch::autograd::FunctionPostHook (#18140) Summary: They are called as (outputs, inputs) and were named (inputs, outputs). Possible follow up fix is to make the outputs argument an lvalue to allow for calling multiple post hooks without ever copying outputs vector. It looks like the copy is now forced because the hook takes a const reference as input and returns an value. This would change the prototype of the function, so needs further discussion. Pull Request resolved: https://github.com/pytorch/pytorch/pull/18140 Differential Revision: D14684498 Pulled By: pietern fbshipit-source-id: 1bd3ddbdd1ff7fe0a18241de5a9ec745a4e7ef07 --- torch/csrc/autograd/function_hook.h | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/torch/csrc/autograd/function_hook.h b/torch/csrc/autograd/function_hook.h index f3cf5b2..33970c1 100644 --- a/torch/csrc/autograd/function_hook.h +++ b/torch/csrc/autograd/function_hook.h @@ -16,7 +16,9 @@ struct FunctionPreHook { struct FunctionPostHook { virtual ~FunctionPostHook() = default; - virtual variable_list operator()(const variable_list& grad_input, const variable_list& grad_output) = 0; + virtual variable_list operator()( + const variable_list& outputs /* grad_inputs */, + const variable_list& inputs /* grad_outputs */) = 0; }; }} // namespace torch::autograd -- 2.7.4