pytorch
7a6808c5 - build: support DNNL_GRAPH_CPU_RUNTIME=TBB (#87512)

Commit
2 years ago
build: support DNNL_GRAPH_CPU_RUNTIME=TBB (#87512) 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 https://github.com/pytorch/pytorch/issues/87511 Pull Request resolved: https://github.com/pytorch/pytorch/pull/87512 Approved by: https://github.com/jgong5, https://github.com/ashokei, https://github.com/malfet
Author
Committer
Parents
Loading