experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Removes potential crash from GraphMgr.

Browse Source 
Change: 144020788
This commit is contained in:
A. Unique TensorFlower 2017-01-09 16:08:04 -08:00 committed by TensorFlower Gardener
parent ee79ebd9bb
commit bfa0fa079d
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tensorflow/core/distributed_runtime/graph_mgr.cc
View File

@ -153,7 +153,12 @@ Status GraphMgr::InitItem(const string& session, const GraphDef& gdef,
// Find the device. // Find the device.
s = worker_env_->device_mgr->LookupDevice(device_name, &unit->device); s = worker_env_->device_mgr->LookupDevice(device_name, &unit->device);
if (!s.ok()) break; if (!s.ok()) {
// Remove the empty unit from the item as the item destructor wants all
// units to have valid devices.
item->units.pop_back();
break;
}
// Construct the subgraph. // Construct the subgraph.
Graph* subgraph = new Graph(item->lib_def); Graph* subgraph = new Graph(item->lib_def);