Merge pull request #12100 from sfraczek/sfraczek/top-k-reserve

luotao1 · web-flow · commit 3a769b9b9032 · 2018-07-13T16:18:14.000+08:00
reserve vector space before loop in top-k
diff --git a/paddle/fluid/operators/top_k_op.h b/paddle/fluid/operators/top_k_op.h
@@ -60,6 +60,7 @@ class TopkKernel : public framework::OpKernel<T> {
 #endif
     for (size_t i = 0; i < row; i++) {
       std::vector<std::pair<T, size_t>> vec;
+      vec.reserve(col);
       for (size_t j = 0; j < col; j++) {
         vec.push_back(std::pair<T, size_t>(eg_input(i, j), j));
       }

Original file line number	Diff line number	Diff line change
`@@ -60,6 +60,7 @@ class TopkKernel : public framework::OpKernel<T> {`
`60`	`60`	`#endif`
`61`	`61`	`for (size_t i = 0; i < row; i++) {`
`62`	`62`	`std::vector<std::pair<T, size_t>> vec;`
	`63`	`+ vec.reserve(col);`
`63`	`64`	`for (size_t j = 0; j < col; j++) {`
`64`	`65`	`vec.push_back(std::pair<T, size_t>(eg_input(i, j), j));`
`65`	`66`	`}`