-
Notifications
You must be signed in to change notification settings - Fork 21.4k
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
build: support DNNL_GRAPH_CPU_RUNTIME=TBB #87512
build: support DNNL_GRAPH_CPU_RUNTIME=TBB #87512
Conversation
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/87512
Note: Links to docs will display an error until the docs builds have been completed. ✅ No Failures, 2 PendingAs of commit 8d6a1a3: This comment was automatically generated by Dr. CI and updates every 15 minutes. |
…ean-cho/pytorch into minjean/mkldnn_tbb_support
Hi @malfet @seemethere, could you please help review this PR? Thanks. |
@pytorchbot merge |
Merge startedYour change will be merged once all checks pass (ETA 0-4 Hours). Learn more about merging in the wiki. Questions? Feedback? Please reach out to the PyTorch DevX Team |
Hey @min-jean-cho. |
Force set cmake `DNNL_GRAPH_CPU_RUNTIME` as `MKLDNN_CPU_RUNTIME` to overwrite [`set(DNNL_GRAPH_CPU_RUNTIME "OMP")`](https://github.com/oneapi-src/oneDNN/blob/d19d0f795c60695bd32f894c6f01771b2dfbe24d/cmake/options.cmake#L65-L67), enabling user-specified `MKLDNN_CPU_RUNTIME` values (`OMP` (default), `TBB`) for `DNNL_GRAPH_CPU_RUNTIME`. Fixes pytorch#87511 Pull Request resolved: pytorch#87512 Approved by: https://github.com/jgong5, https://github.com/ashokei, https://github.com/malfet
Force set cmake `DNNL_GRAPH_CPU_RUNTIME` as `MKLDNN_CPU_RUNTIME` to overwrite [`set(DNNL_GRAPH_CPU_RUNTIME "OMP")`](https://github.com/oneapi-src/oneDNN/blob/d19d0f795c60695bd32f894c6f01771b2dfbe24d/cmake/options.cmake#L65-L67), enabling user-specified `MKLDNN_CPU_RUNTIME` values (`OMP` (default), `TBB`) for `DNNL_GRAPH_CPU_RUNTIME`. Fixes pytorch#87511 Pull Request resolved: pytorch#87512 Approved by: https://github.com/jgong5, https://github.com/ashokei, https://github.com/malfet
Force set cmake `DNNL_GRAPH_CPU_RUNTIME` as `MKLDNN_CPU_RUNTIME` to overwrite [`set(DNNL_GRAPH_CPU_RUNTIME "OMP")`](https://github.com/oneapi-src/oneDNN/blob/d19d0f795c60695bd32f894c6f01771b2dfbe24d/cmake/options.cmake#L65-L67), enabling user-specified `MKLDNN_CPU_RUNTIME` values (`OMP` (default), `TBB`) for `DNNL_GRAPH_CPU_RUNTIME`. Fixes pytorch#87511 Pull Request resolved: pytorch#87512 Approved by: https://github.com/jgong5, https://github.com/ashokei, https://github.com/malfet
Force set cmake
DNNL_GRAPH_CPU_RUNTIME
asMKLDNN_CPU_RUNTIME
to overwriteset(DNNL_GRAPH_CPU_RUNTIME "OMP")
, enabling user-specifiedMKLDNN_CPU_RUNTIME
values (OMP
(default),TBB
) forDNNL_GRAPH_CPU_RUNTIME
.Fixes #87511