From b06ae1990e46c91b9698bc43c8b828b7681256c3 Mon Sep 17 00:00:00 2001 From: =?utf8?q?=D0=94=D0=B8=D0=BB=D1=88=D0=BE=D0=B4=D0=B6=D0=BE=D0=BD=20?= =?utf8?q?=D0=A3=D0=BC=D1=80=D0=BE=D0=BD=D1=85=D0=BE=D0=BD=D0=BE=D0=B2?= =?utf8?q?=D0=B8=D1=87=20=D0=9F=D0=BE=D1=88=D1=88=D0=BE=D0=B5=D0=B2/AI=20T?= =?utf8?q?ools=20Lab=20/SRR/Engineer/=EC=82=BC=EC=84=B1=EC=A0=84=EC=9E=90?= Date: Mon, 25 Mar 2019 12:33:57 +0300 Subject: [PATCH] Redo TensorBuilderSet as unordered_set (#4836) Redo TensorBuilderSet as unordered_set since we don't care if it is sorted or not Signed-off-by: Poshshoev Dilshodzhon --- runtimes/neurun/core/include/backend/ITensorBuilder.h | 4 ++-- runtimes/neurun/core/src/compiler/Compiler.cc | 2 +- runtimes/neurun/core/src/exec/DataflowExecutor.h | 1 - 3 files changed, 3 insertions(+), 4 deletions(-) diff --git a/runtimes/neurun/core/include/backend/ITensorBuilder.h b/runtimes/neurun/core/include/backend/ITensorBuilder.h index 2f7165f..514ec3a 100644 --- a/runtimes/neurun/core/include/backend/ITensorBuilder.h +++ b/runtimes/neurun/core/include/backend/ITensorBuilder.h @@ -63,7 +63,7 @@ struct ITensorBuilder } // namespace backend } // namespace neurun -#include +#include #include namespace neurun @@ -71,7 +71,7 @@ namespace neurun namespace backend { -using TensorBuilderSet = std::set>; +using TensorBuilderSet = std::unordered_set>; } // namespace backend } // namespace neurun diff --git a/runtimes/neurun/core/src/compiler/Compiler.cc b/runtimes/neurun/core/src/compiler/Compiler.cc index edef0d0..270f562 100644 --- a/runtimes/neurun/core/src/compiler/Compiler.cc +++ b/runtimes/neurun/core/src/compiler/Compiler.cc @@ -159,7 +159,7 @@ std::shared_ptr Compiler::createDataflowExecutor(graph::Graph & stages[index] = stage_gen->generate(node); }); - std::unordered_set> tensor_builders; + backend::TensorBuilderSet tensor_builders; model.operands().iterate( [&](const model::operand::Index &ind, const model::operand::Object &obj) { diff --git a/runtimes/neurun/core/src/exec/DataflowExecutor.h b/runtimes/neurun/core/src/exec/DataflowExecutor.h index 3677162..f9a76f3 100644 --- a/runtimes/neurun/core/src/exec/DataflowExecutor.h +++ b/runtimes/neurun/core/src/exec/DataflowExecutor.h @@ -19,7 +19,6 @@ #include #include -#include #include #include "FunctionSequence.h" -- 2.7.4