small fix

7 years ago · a9217031ba
parent 509b6835be
commit a9217031ba
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"