lzhengning
/
JittorMirror
mirror of https://github.com/Jittor/Jittor
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

add grad tracer

This commit is contained in:
Dun Liang 2020-12-23 23:03:00 +08:00
parent 37b70ddf2c
commit 70ea7202da
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/pybind/py_var_tracer.cc
View File

@ -191,6 +191,7 @@ void TraceData::record_node(Node* node, bool record_stack) {
auto iter = trace_data.id_map.find(trace_grad_op);
data.stacks.emplace_back(Stack{"grad", "Grad", "", 0});
if (iter != trace_data.id_map.end()) {
data.attrs["grad_op_id"] = S(iter->second);
auto& prev_stack = trace_data.node_data[iter->second].stacks;
for (auto& s : prev_stack)
data.stacks.push_back(s);