From 5ca15816803f0ca5d59b9a90fd11f72d4605e69d Mon Sep 17 00:00:00 2001
From: Yanzhan Yang <yangyanzhan@gmail.com>
Date: Wed, 3 Jul 2019 10:09:20 +0800
Subject: [PATCH] fix fast check (#1720)

* fix fast check

* fix style
---
 tools/python/fluidtools/run.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tools/python/fluidtools/run.py b/tools/python/fluidtools/run.py
index 291a7729a8..804062b76f 100644
--- a/tools/python/fluidtools/run.py
+++ b/tools/python/fluidtools/run.py
@@ -225,6 +225,7 @@ def save_all_op_output(feed_kv=None):
     fetch_names = []
     for fetch in fetches:
         fetch_names.append(fetch.name)
+    feed_names = feeds
     for i in range(len(ops)):
         op = ops[i]
         var_name = None
@@ -235,7 +236,7 @@ def save_all_op_output(feed_kv=None):
         if "sequence_pool" in var_name:
             continue
         if fast_check:
-            if var_name not in fetch_names:
+            if var_name not in fetch_names and var_name not in feed_names:
                 continue
         try:
             data = get_var_data(var_name, feed_kv=feed_kv).flatten().tolist()
-- 
GitLab