From 278f55bcef26b68be57882225d29d8ed268557eb Mon Sep 17 00:00:00 2001 From: mozga-intel Date: Mon, 28 May 2018 09:26:00 +0200 Subject: [PATCH] Support for mklml version of code --- paddle/fluid/operators/top_k_op.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paddle/fluid/operators/top_k_op.h b/paddle/fluid/operators/top_k_op.h index b31fee22a..7ddb82ef6 100644 --- 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 { // 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++) { -- GitLab