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

Trace C functions

Browse Source
This commit is contained in:
Adam Paszke
2017-08-02 14:40:11 -07:00
committed by Soumith Chintala
parent bdcbbeaf68
commit 1c4538e017
18 changed files with 126 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
torch/csrc/autograd/python_function.cpp
View File

@ -211,6 +211,10 @@ auto PyFunction::name() -> std::string {
return std::string(Py_TYPE(f)->tp_name);
}
auto PyFunction::getSharedPtr() -> std::shared_ptr<Function> {
return THPFunction_asFunction((THPFunction*)obj);
}
}} // namespace torch::autograd
// Traverse and clear are required for supporting Python's GC cycle handling.
@ -322,6 +326,8 @@ static void _mark_dirty(THPFunction *self, t2var_type &t2var,
"variables, but detected that there are %d objects sharing it",
v_counter.var_refcnt());
v_counter++;
// In-place modifications invalidate the trace
variable->cdata->tracing_state.reset();
}
// We're not going to ever need this so let's remove references now
Py_DECREF(self->dirty_tensors);