mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-21 13:44:15 +08:00
da6b8a905a
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/28605 This was added because std::to_string isn't available in libstc++ on Android. Use it in more places to get the PyTorch Android build working with libstdc++. Test Plan: Internal android build. Reviewed By: jerryzh168 Differential Revision: D18099520 fbshipit-source-id: 17a2b617c2d21deadd0fdac1db849823637981fc
44 lines
1.2 KiB
C++
44 lines
1.2 KiB
C++
#include <torch/csrc/autograd/cpp_hook.h>
|
|
#include <torch/csrc/autograd/variable.h>
|
|
#include <torch/csrc/autograd/custom_function.h>
|
|
|
|
namespace {
|
|
using torch::autograd::Variable;
|
|
void check_single_result (Variable value, Variable result, std::string hook_name) {
|
|
if (!value.defined()) {
|
|
throw std::runtime_error("can't replace a empty gradient with a non-empty value");
|
|
}
|
|
torch::autograd::check_variable_result(value, result, hook_name);
|
|
}
|
|
}
|
|
|
|
namespace torch { namespace autograd {
|
|
|
|
CppFunctionPreHook::CppFunctionPreHook(const std::shared_ptr<hooks_list> &hooks, int value_idx)
|
|
: hooks_(hooks)
|
|
, value_idx_(value_idx)
|
|
{}
|
|
|
|
variable_list CppFunctionPreHook::operator()(const variable_list& values) {
|
|
auto value = values[value_idx_];
|
|
for (unsigned i = 0; i < hooks_->size(); ++i) {
|
|
auto &hook = (*hooks_)[i];
|
|
if (!hook) {
|
|
// hook was removed
|
|
continue;
|
|
}
|
|
auto res = hook(value);
|
|
if (!res.defined()) {
|
|
// Don't change gradient
|
|
continue;
|
|
}
|
|
check_single_result(value, res, c10::to_string(i));
|
|
value = res;
|
|
}
|
|
variable_list results(values);
|
|
results[value_idx_] = value;
|
|
return results;
|
|
}
|
|
|
|
}} // namespace torch::autograd
|