From 1ba443a4fea5a3161ff64b50d4b5e22aac859f4d Mon Sep 17 00:00:00 2001
From: Chang Lan <changlan9@gmail.com>
Date: Thu, 6 Dec 2018 12:33:00 -0800
Subject: [PATCH] Use unsafe_get in nnvm (#2247)
---
3rdparty/dmlc-core | 2 +-
nnvm/include/nnvm/base.h | 3 +++
nnvm/include/nnvm/graph.h | 2 +-
3 files changed, 5 insertions(+), 2 deletions(-)
diff --git a/3rdparty/dmlc-core b/3rdparty/dmlc-core
index 4d49691f1..519d013a2 160000
--- a/3rdparty/dmlc-core
+++ b/3rdparty/dmlc-core
@@ -1 +1 @@
-Subproject commit 4d49691f1a9d944c3b0aa5e63f1db3cad1f941f8
+Subproject commit 519d013a213c0c447a971f51219473ef564d2348
diff --git a/nnvm/include/nnvm/base.h b/nnvm/include/nnvm/base.h
index 449bd2f46..39ff70093 100644
--- a/nnvm/include/nnvm/base.h
+++ b/nnvm/include/nnvm/base.h
@@ -25,6 +25,9 @@ using dmlc::array_view;
/*!\brief getter function of any type */
using dmlc::get;
+/*!\brief "unsafe" getter function of any type */
+using dmlc::unsafe_get;
+
} // namespace nnvm
// describe op registration point
diff --git a/nnvm/include/nnvm/graph.h b/nnvm/include/nnvm/graph.h
index 3f8a2a364..93612ccb9 100644
--- a/nnvm/include/nnvm/graph.h
+++ b/nnvm/include/nnvm/graph.h
@@ -229,7 +229,7 @@ inline const T& Graph::GetAttr(const std::string& attr_name) const {
auto it = attrs.find(attr_name);
CHECK(it != attrs.end())
<< "Cannot find attribute " << attr_name << " in the graph";
- return nnvm::get<T>(*it->second);
+ return nnvm::unsafe_get<T>(*it->second);
}
inline bool Graph::HasAttr(const std::string& attr_name) const {
--
GitLab