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

polish trace_py_var

This commit is contained in:
Dun Liang 2020-12-28 14:42:50 +08:00
parent 1acda62f59
commit 4af3f7a0d2
3 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/executor.cc
View File

@ -449,7 +449,7 @@ void Executor::run_sync(vector<Var*> vars, bool device_sync) {
last_is_cuda = is_cuda;
op->do_run_after_prepare(jkl);
// record trace data
if (PREDICT_BRANCH_NOT_TAKEN(trace_py_var==2)) {
if (PREDICT_BRANCH_NOT_TAKEN(trace_py_var>=2)) {
trace_data.record_execution(op, is_fused_op, jkl);
#ifdef HAS_CUDA
if (use_cuda)

4
src/profiler/profiler.cc
View File

@ -83,7 +83,7 @@ static string get_stack_info(Op* op) {
stack_info += kv.first;
stack_info += '\n';
}
if (trace_py_var == 2) {
if (trace_py_var >= 3) {
std::stringstream ss;
ss << "input from:\n";
for (auto& vi : fop->vars) {
@ -103,7 +103,7 @@ static string get_stack_info(Op* op) {
} else {
stack_info += _get_stack_info(op);
stack_info += '\n';
if (trace_py_var == 2) {
if (trace_py_var >= 3) {
std::stringstream ss;
ss << "input from:\n";
for (auto v : op->inputs()) {

6
src/pybind/py_var_tracer.cc
View File

@ -95,7 +95,7 @@ static vector<Stack> get_stack_info() {
int i=n;
while (i) frames[--i] = frame, frame = frame->f_back;
PyObject* prev_obj = nullptr;
if (trace_py_var == 2) {
if (trace_py_var >= 3) {
// trace raw stack
auto start = std::max(0, n-5);
for (int i=start; i<n; i++) {
@ -185,7 +185,7 @@ void TraceData::record_node(Node* node, bool record_stack) {
NodeData data;
data.id = node_data_cnt++;
id_map[node] = data.id;
if (!node->is_var() || trace_py_var==2) {
if (!node->is_var() || trace_py_var>=3) {
if (record_stack) {
if (trace_grad_op) {
auto iter = trace_data.id_map.find(trace_grad_op);
@ -219,7 +219,7 @@ void TraceData::release_node(Node* node) {
return;
auto node_id = iter->second;
id_map.erase(node);
if (trace_py_var == 1) {
if (trace_py_var >= 1) {
node_data.erase(node_id);
}
}