refine frontend (PaddlePaddle#174)

thisjiang · Aug 14, 2020 · 4ace494 · 4ace494
1 parent 1a2b6ca
commit 4ace494
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 35 deletions.
diff --git a/cinn/frontend/CMakeLists.txt b/cinn/frontend/CMakeLists.txt
@@ -1,6 +1,5 @@
 set(srcs
   syntax.cc
-  api.cc
   )
 
 cc_test(test_frontend_syntax SRCS syntax_test.cc DEPS core)

diff --git a/cinn/frontend/api.cc b/cinn/frontend/api.cc
diff --git a/cinn/frontend/api.h b/cinn/frontend/api.h
diff --git a/cinn/frontend/syntax.h b/cinn/frontend/syntax.h
@@ -48,18 +48,13 @@ struct _Variable_ : public common::Object {
   static constexpr char* __type_info__ = "cinn_frontend_variable";
 };
 
-struct Variable {
-  explicit Variable(std::string_view id = "") {
-    auto* n = common::make_shared<_Variable_>();
-    n->id   = id.empty() ? common::Context::Global().NewName("var") : id;
-    data_.Reset(n);
+struct Variable : public common::Shared<_Variable_> {
+  explicit Variable(std::string_view id = "") : common::Shared<_Variable_>(common::make_shared<_Variable_>()) {
+    get()->id = id.empty() ? common::Context::Global().NewName("var") : id;
   }
 
-  _Variable_* operator->() { return data_.get(); }
-  const _Variable_* operator->() const { return data_.get(); }
-
- private:
-  common::Shared<_Variable_> data_;
+  _Variable_* operator->() { return get(); }
+  const _Variable_* operator->() const { return get(); }
 };
 
 struct _Instruction_ : public common::Object {