Skip to content

Commit

Permalink
Remove PADDLE_WITH_MKLDNN in custom_op
Browse files Browse the repository at this point in the history
  • Loading branch information
Aurelius84 committed May 14, 2021
1 parent 890f626 commit 296781e
Showing 1 changed file with 0 additions and 4 deletions.
4 changes: 0 additions & 4 deletions python/paddle/utils/cpp_extension/extension_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -469,10 +469,6 @@ def normalize_extension_kwargs(kwargs, use_cuda=False):
########################### -- END -- ###########################

add_compile_flag(extra_compile_args, ['-w']) # disable warning
# Note(Aurelius84): This marco will impact memory layout of `Tensor`.
# We align it automatically with pre-installed Paddle.
if core.is_compiled_with_mkldnn():
add_compile_flag(extra_compile_args, ['-DPADDLE_WITH_MKLDNN'])

if use_cuda:
extra_link_args.append('-lcudart')
Expand Down

1 comment on commit 296781e

@paddle-bot-old
Copy link

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Congratulation! Your pull request passed all required CI. You could ask reviewer(s) to approve and merge. 🎉

Please sign in to comment.