Support for mklml version of code

7 years ago · 278f55bcef
parent e7f0696513
commit 278f55bcef
1 changed files with 1 additions and 1 deletions
--- a/paddle/fluid/operators/top_k_op.h
+++ b/paddle/fluid/operators/top_k_op.h
@ -55,7 +55,7 @@ class TopkKernel : public framework::OpKernel<T> {
    // NOTE: eigen shape doesn't affect paddle tensor.
    eg_input.reshape(flat2dims);

-#ifdef PADDLE_WITH_MKLDNN
+#ifdef PADDLE_WITH_MKLML
 #pragma omp parallel for
 #endif
    for (size_t i = 0; i < row; i++) {