fix analyzer_mm_dnn_tester fails when bs > 1

test=develop
6 years ago · 91408e3122
parent f01c966800
commit 91408e3122
1 changed files with 1 additions and 1 deletions
--- a/paddle/fluid/inference/tests/api/analyzer_mm_dnn_tester.cc
+++ b/paddle/fluid/inference/tests/api/analyzer_mm_dnn_tester.cc
@ -47,7 +47,7 @@ struct DataRecord {
      for (size_t j = 0; j < data.query_data_all.size(); j++) {
        // calculate lod
        data.lod1.push_back(data.lod1.back() + data.query_data_all[j].size());
-        data.lod2.push_back(data.lod2.back() + data.query_data_all[j].size());
+        data.lod2.push_back(data.lod2.back() + data.title_data_all[j].size());
      }
    }
    batch_iter += batch_size;