frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-11-06 09:17:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

Trace fork and join calls

Browse Source 
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/16232

Differential Revision: D13772974

Pulled By: jamesr66a

fbshipit-source-id: b2db370271809e26d3301f8cc98eec567db5e62b
This commit is contained in:
James Reed
2019-01-26 14:38:12 -08:00
committed by Facebook Github Bot
parent 8c81a72e87
commit d1ed0176df
20 changed files with 443 additions and 120 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tools/autograd/templates/python_variable_methods.cpp
View File

@ -149,7 +149,7 @@ static Tensor dispatch_contiguous(const Tensor & self) {
auto node = tracer_state->graph->create(jit::aten::contiguous, /*num_outputs=*/0);
jit::tracer::recordSourceLocation(node);
jit::tracer::addInputs(node, "self", self_);
tracer_state->graph->appendNode(node);
tracer_state->graph->insertNode(node);
jit::tracer::addOutput(node, self_);
}
Py_INCREF(self);