Merge pull request #12774 from panyx0718/infer

small fix
7 years ago · 7c5f08e51d
parent 509b6835be a9217031ba
commit 7c5f08e51d
1 changed files with 4 additions and 0 deletions
--- a/paddle/fluid/framework/ir/graph_traits.h
+++ b/paddle/fluid/framework/ir/graph_traits.h
@ -12,7 +12,11 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.

+#pragma once
+
 #include <stack>
+#include <vector>
+
 #include "paddle/fluid/framework/ir/graph.h"
 #include "paddle/fluid/framework/ir/node.h"