Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[INTEL MKL] Refactoring code related to compiling oneDNN with threadpool. #47279

Merged

Conversation

agramesh1
Copy link
Contributor

Refactoring code to replace the macro ENABLE_MKLDNN_THREADPOOL with ENABLE_ONEDNN_OPENMP. Now, the oneDNN based kernels will be compiled to use Eigen threadpool by default unless ENABLE_ONEDNN_OPENMP is defined. PR also includes some minor cleanup to remove unused mkl_opensource build option.

@google-ml-butler google-ml-butler bot added the size:M CL Change Size: Medium label Feb 20, 2021
@google-cla google-cla bot added the cla: yes label Feb 20, 2021
@gbaned gbaned self-assigned this Feb 22, 2021
@gbaned gbaned added the comp:mkl MKL related issues label Feb 22, 2021
Copy link
Member

@penpornk penpornk left a comment

Choose a reason for hiding this comment

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

@cuixiaom Thank you for the heads up!
@agramesh1 Sorry for the delay! I thought this one was not urgent based on the title. I have a few nits.

tensorflow/core/kernels/mkl/mkl_fused_ops_test.cc Outdated Show resolved Hide resolved
third_party/mkl_dnn/build_defs.bzl Outdated Show resolved Hide resolved
third_party/mkl_dnn/build_defs.bzl Show resolved Hide resolved
Copy link
Member

@penpornk penpornk left a comment

Choose a reason for hiding this comment

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

Thank you and have a great weekend! :)

third_party/mkl_dnn/build_defs.bzl Show resolved Hide resolved
@google-ml-butler google-ml-butler bot added kokoro:force-run Tests on submitted change ready to pull PR ready for merge process labels Feb 27, 2021
@google-ml-butler google-ml-butler bot removed the ready to pull PR ready for merge process label Feb 27, 2021
@kokoro-team kokoro-team removed the kokoro:force-run Tests on submitted change label Feb 27, 2021
@agramesh1
Copy link
Contributor Author

Thank you and have a great weekend! :)

@penpornk thanks. There was a conflict I resolved just after you approved. Not sure if the changes registered before the testing started.

Copy link
Member

@penpornk penpornk left a comment

Choose a reason for hiding this comment

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

I see. Approving again then. :)

@google-ml-butler google-ml-butler bot added kokoro:force-run Tests on submitted change ready to pull PR ready for merge process labels Feb 27, 2021
@kokoro-team kokoro-team removed the kokoro:force-run Tests on submitted change label Feb 27, 2021
@google-ml-butler google-ml-butler bot removed the ready to pull PR ready for merge process label Mar 1, 2021
@google-ml-butler google-ml-butler bot added kokoro:force-run Tests on submitted change ready to pull PR ready for merge process labels Mar 2, 2021
@kokoro-team kokoro-team removed the kokoro:force-run Tests on submitted change label Mar 2, 2021
@agramesh1
Copy link
Contributor Author

@penpornk thanks for approving again.

@agramesh1
Copy link
Contributor Author

Hi @penpornk The internal Google checks failed for this PR, can you provide the logs for the failures?

@penpornk
Copy link
Member

penpornk commented Mar 5, 2021

@agramesh1 Thank you for the heads up! I'll fix this internally.

@copybara-service copybara-service bot merged commit 4095c30 into tensorflow:master Mar 5, 2021
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
cla: yes comp:mkl MKL related issues ready to pull PR ready for merge process size:M CL Change Size: Medium
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants