From 5c224c71f0a0b7a07bacacf45051b27365e4b88c Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Tue, 14 Jul 2020 18:52:17 +0800 Subject: [PATCH] refactor(mge/imperative): drop ComputingGraphImpl in proxy graph GitOrigin-RevId: a6e6f2e76bba002555d20951d1590d0a532b1b10 --- src/core/include/megbrain/graph/static_infer.h | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/src/core/include/megbrain/graph/static_infer.h b/src/core/include/megbrain/graph/static_infer.h index b79ed827..93859bde 100644 --- a/src/core/include/megbrain/graph/static_infer.h +++ b/src/core/include/megbrain/graph/static_infer.h @@ -14,6 +14,11 @@ #include "megbrain/tensor.h" namespace mgb { + +namespace imperative { + class ProxyGraph; +} // namespace imperative + namespace cg { class VarNode; @@ -50,6 +55,7 @@ namespace static_infer { const DeviceTensorND *m_value = nullptr; friend class StaticInferManagerImpl; + friend class imperative::ProxyGraph; public: /*! -- GitLab