From 5b31bd7ae3345e555493c26b1b941f6dfcfd2387 Mon Sep 17 00:00:00 2001
From: Yu Yang <yuyang18@baidu.com>
Date: Tue, 27 Jun 2017 11:29:30 +0800
Subject: [PATCH] Add `Variable::IsType`

Make user can check what type is in variable
---
 paddle/framework/variable.h | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/paddle/framework/variable.h b/paddle/framework/variable.h
index b33e10e6820..72c4a7a2a1d 100644
--- a/paddle/framework/variable.h
+++ b/paddle/framework/variable.h
@@ -25,21 +25,24 @@ class Variable {
  public:
   template <typename T>
   const T& Get() const {
-    PADDLE_ASSERT(holder_ != nullptr);
-    PADDLE_ASSERT(std::type_index(typeid(T)) ==
-                  std::type_index(holder_->Type()));
+    PADDLE_ASSERT(IsType<T>());
     return *static_cast<const T*>(holder_->Ptr());
   }
 
   template <typename T>
   T* GetMutable() {
-    if (holder_ == nullptr ||
-        std::type_index(typeid(T)) != std::type_index(holder_->Type())) {
+    if (!IsType<T>()) {
       holder_.reset(new PlaceholderImpl<T>(new T()));
     }
     return static_cast<T*>(holder_->Ptr());
   }
 
+  template <typename T>
+  bool IsType() const {
+    return holder_ != nullptr &&
+           std::type_index(typeid(T)) == std::type_index(holder_->Type());
+  }
+
  private:
   struct Placeholder {
     virtual ~Placeholder() {}
-- 
GitLab