frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 12:54:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

Revert "[Submodule] Remove deprecated USE_TBB option and TBB submodule (#127051)"

Browse Source 
This reverts commit 4fdbaa794f9d5af2f171f772a51cb710c51c925f.

Reverted https://github.com/pytorch/pytorch/pull/127051 on behalf of https://github.com/PaliC due to This PR needs to be synced using the import button as there is a bug in our diff train ([comment](https://github.com/pytorch/pytorch/pull/127051#issuecomment-2136428735))
This commit is contained in:
PyTorch MergeBot
2024-05-29 03:02:35 +00:00
parent 7a506dd005
commit cdbb2c9acc
34 changed files with 863 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
defs.bzl
View File

@ -64,6 +64,8 @@ def get_cpu_parallel_backend_flags():
defs = []
if parallel_backend == "openmp":
defs.append("-DAT_PARALLEL_OPENMP_FBCODE=1")
elif parallel_backend == "tbb":
defs.append("-DAT_PARALLEL_NATIVE_TBB_FBCODE=1")
elif parallel_backend == "native":
defs.append("-DAT_PARALLEL_NATIVE_FBCODE=1")
else: