Support for mklml version of code

278f55bc · mozga-intel · e7f06965 · 278f55bc
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

paddle/fluid/operators/top_k_op.h paddle/fluid/operators/top_k_op.h +1 -1

未找到文件。
--- 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++) {