saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-31 23:27:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add check for nullptr in PlannerImpl::FindReusableTensor(). (#3619)

Browse source
This commit is contained in:
edgchen1 2020-04-22 20:18:29 -07:00 committed by GitHub
parent 0e12d05cd2
commit 2dd4f7e96b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
onnxruntime/core/framework/allocation_planner.cc
View file

@ -330,6 +330,11 @@ class PlannerImpl {
for (auto it = freelist_.begin(); it != freelist_.end(); ++it) {
size_t reusable = static_cast<size_t>(it->ml_value);
const onnxruntime::NodeArg* p_node_arg = ort_value_info_.at(reusable).p_def_site;
if (!p_node_arg) {
// TODO this should be an error case, needs more investigation
// https://msdata.visualstudio.com/Vienna/_workitems/edit/724826/
continue;
}
auto& available_memory_info = AllocPlan(p_node_arg->Name()).location;
if (!(available_memory_info == required_memory_info)) continue;
auto p_available_buffer_shape = context_.GetShape(*p_node_arg);