diff --git a/benchmark/fluid/models/mnist.py b/benchmark/fluid/models/mnist.py
index d903a834ec0be8e51f7e1378f4a2e9e20bae8583..8e740dc6896b7eeeb82170aa13d32987c4df5c48 100644
--- a/benchmark/fluid/models/mnist.py
+++ b/benchmark/fluid/models/mnist.py
@@ -75,7 +75,8 @@ def get_model(args):
             shapes=[[-1, 1, 28, 28], (-1, 1)],
             lod_levels=[0, 0],
             dtypes=["float32", "int64"],
-            thread_num=args.gpus)
+            thread_num=args.gpus,
+            pass_num=args.pass_num)
         data_file = fluid.layers.double_buffer(
             fluid.layers.batch(
                 data_file, batch_size=args.batch_size))
diff --git a/benchmark/fluid/models/resnet.py b/benchmark/fluid/models/resnet.py
index 88451064fcc150b083c77565de30657b1886ab81..2ee2b5be09bfcc2e7fcec7eb2f80e28e4e75ab3d 100644
--- a/benchmark/fluid/models/resnet.py
+++ b/benchmark/fluid/models/resnet.py
@@ -157,7 +157,8 @@ def get_model(args):
             shapes=[[-1] + dshape, (-1, 1)],
             lod_levels=[0, 0],
             dtypes=["float32", "int64"],
-            thread_num=args.gpus)
+            thread_num=args.gpus,
+            pass_num=args.pass_num)
         data_file = fluid.layers.double_buffer(
             fluid.layers.batch(
                 data_file, batch_size=args.batch_size))
diff --git a/benchmark/fluid/models/vgg.py b/benchmark/fluid/models/vgg.py
index cb0dc977634429178c441b1d2777f865c749c2db..6092cdeb884b3a9b60a3bcf20b022f2b0685e6aa 100644
--- a/benchmark/fluid/models/vgg.py
+++ b/benchmark/fluid/models/vgg.py
@@ -75,7 +75,8 @@ def get_model(args):
             shapes=[[-1] + data_shape, (-1, 1)],
             lod_levels=[0, 0],
             dtypes=["float32", "int64"],
-            thread_num=args.gpus)
+            thread_num=args.gpus,
+            pass_num=args.pass_num)
         data_file = fluid.layers.double_buffer(
             fluid.layers.batch(
                 data_file, batch_size=args.batch_size))