Skip to content

Commit 278f55b

Browse files
committed
Support for mklml version of code
1 parent e7f0696 commit 278f55b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

paddle/fluid/operators/top_k_op.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,7 @@ class TopkKernel : public framework::OpKernel<T> {
5555
// NOTE: eigen shape doesn't affect paddle tensor.
5656
eg_input.reshape(flat2dims);
5757

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

0 commit comments

Comments
 (0)