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

[BE]: Use make_unique (#126966)

Browse Source 
Adds make_unique in places

Pull Request resolved: https://github.com/pytorch/pytorch/pull/126966
Approved by: https://github.com/Skylion007
This commit is contained in:
cyy
2024-05-23 17:39:45 +00:00
committed by PyTorch MergeBot
parent 264155a8d7
commit e5db6758c8
10 changed files with 25 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
torch/csrc/autograd/python_cpp_function.cpp
View File

@ -175,9 +175,8 @@ PyObject* THPCppFunction_register_hook_dict(PyObject* self, PyObject* _var) {
}
auto var = (THPVariable*)_var;
auto& fn = *((THPCppFunction*)self)->cdata;
std::unique_ptr<FunctionPreHook> hook(new PyFunctionTensorPreHook(
fn.add_tensor_pre_hook(std::make_unique<PyFunctionTensorPreHook>(
var->backward_hooks, THPVariable_Unpack(var).output_nr()));
fn.add_tensor_pre_hook(std::move(hook));
Py_RETURN_NONE;
}
@ -346,8 +345,7 @@ PyObject* registerFunctionHook(Node& fn, PyObject* hook) {
}
if (dict == Py_None) {
dict = PyTuple_GET_ITEM(res.get(), 0);
std::unique_ptr<FunctionPostHook> hook(new PyFunctionPostHook(dict));
fn.add_post_hook(std::move(hook));
fn.add_post_hook(std::make_unique<PyFunctionPostHook>(dict));
}
PyObject* handle = PyTuple_GET_ITEM(res.get(), 1);
@ -370,8 +368,7 @@ PyObject* registerFunctionPreHook(Node& fn, PyObject* hook) {
}
if (dict == Py_None) {
dict = PyTuple_GET_ITEM(res.get(), 0);
std::unique_ptr<FunctionPreHook> hook(new PyFunctionPreHook(dict));
fn.add_pre_hook(std::move(hook));
fn.add_pre_hook(std::make_unique<PyFunctionPreHook>(dict));
}
PyObject* handle = PyTuple_GET_ITEM(res.get(), 1);