From 66888fcb1a52c514aac6ef7ebe175d8994a12e55 Mon Sep 17 00:00:00 2001
From: Tong Zhigao <tongzhigao@pingcap.com>
Date: Wed, 8 Apr 2020 13:21:56 +0800
Subject: [PATCH] Fix: should set batch flag in DAGQuerySource (#615)

Signed-off-by: Tong Zhigao <tongzhigao@pingcap.com>
---
 dbms/src/Flash/Coprocessor/DAGDriver.cpp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/dbms/src/Flash/Coprocessor/DAGDriver.cpp b/dbms/src/Flash/Coprocessor/DAGDriver.cpp
index bca325df884..87c1376892a 100644
--- a/dbms/src/Flash/Coprocessor/DAGDriver.cpp
+++ b/dbms/src/Flash/Coprocessor/DAGDriver.cpp
@@ -60,7 +60,7 @@ void DAGDriver<batch>::execute()
 try
 {
     DAGContext dag_context(dag_request.executors_size());
-    DAGQuerySource dag(context, dag_context, regions, dag_request);
+    DAGQuerySource dag(context, dag_context, regions, dag_request, batch);
 
     BlockIO streams = executeQuery(dag, context, internal, QueryProcessingStage::Complete);
     if (!streams.in || streams.out)
@@ -140,7 +140,7 @@ catch (...)
 }
 
 template <bool batch>
-void DAGDriver<batch>::recordError(Int32 err_code , const String & err_msg)
+void DAGDriver<batch>::recordError(Int32 err_code, const String & err_msg)
 {
     if constexpr (batch)
     {