[neurun] Use Index as key for unordered_set (#2590)

This commit revises `unordered_set`s to use `graph::Index` as key
which used `int` as key because hash function was not exist.

Signed-off-by: Hanjoung Lee <hanjoung.lee@samsung.com>

diff --git a/runtimes/neurun/src/backend/acl_cl/TensorBuilder.cc b/runtimes/neurun/src/backend/acl_cl/TensorBuilder.cc
index 4ff2efd..880ff2d 100644 (file)
--- a/runtimes/neurun/src/backend/acl_cl/TensorBuilder.cc
+++ b/runtimes/neurun/src/backend/acl_cl/TensorBuilder.cc
@@ -20,7 +20,7 @@ void TensorBuilder::mark(const ::neurun::graph::operand::Index &ind)
 {
   assert(_tensors.size() == 0);
 
-  _inds.insert(ind.asInt());
+  _inds.insert(ind);
 }
 
 void TensorBuilder::prepare(codegen::Plan &plan,

diff --git a/runtimes/neurun/src/backend/acl_cl/TensorBuilder.h b/runtimes/neurun/src/backend/acl_cl/TensorBuilder.h
index c593416..e3c8664 100644 (file)
--- a/runtimes/neurun/src/backend/acl_cl/TensorBuilder.h
+++ b/runtimes/neurun/src/backend/acl_cl/TensorBuilder.h
@@ -30,7 +30,7 @@ public:
   std::shared_ptr<::arm_compute::CLTensor> at(const ::neurun::graph::operand::Index &ind);
 
 private:
-  std::unordered_set<int> _inds;
+  std::unordered_set<graph::operand::Index> _inds;
   std::unordered_map<int, std::shared_ptr<::arm_compute::CLTensor>> _tensors;
 };
 

diff --git a/runtimes/neurun/src/backend/cpu/TensorBuilder.cc b/runtimes/neurun/src/backend/cpu/TensorBuilder.cc
index f25a954..3b7fa80 100644 (file)
--- a/runtimes/neurun/src/backend/cpu/TensorBuilder.cc
+++ b/runtimes/neurun/src/backend/cpu/TensorBuilder.cc
@@ -20,7 +20,7 @@ void TensorBuilder::mark(const ::neurun::graph::operand::Index &ind)
 {
   assert(_tensors.size() == 0);
 
-  _inds.insert(ind.asInt());
+  _inds.insert(ind);
 }
 
 void TensorBuilder::prepare(codegen::Plan &plan,

diff --git a/runtimes/neurun/src/backend/cpu/TensorBuilder.h b/runtimes/neurun/src/backend/cpu/TensorBuilder.h
index 80fa12f..37f3581 100644 (file)
--- a/runtimes/neurun/src/backend/cpu/TensorBuilder.h
+++ b/runtimes/neurun/src/backend/cpu/TensorBuilder.h
@@ -6,6 +6,7 @@
 
 #include "backend/ITensorBuilder.h"
 #include "backend/cpu/operand/Tensor.h"
+#include "graph/operand/Index.h"
 
 namespace neurun
 {
@@ -29,7 +30,7 @@ public:
   std::shared_ptr<operand::Tensor> at(const ::neurun::graph::operand::Index &ind);
 
 private:
-  std::unordered_set<int> _inds;
+  std::unordered_set<graph::operand::Index> _inds;
   std::unordered_map<int, std::shared_ptr<operand::Tensor>> _tensors;
 };