OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

[PyTorch] Update Bazel build for TensorPipe (#54416)

Browse Source 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/54416

Once D27230990 lands, we'll need this for TensorPipe to be built with Bazel.
ghstack-source-id: 124512701

Test Plan: None for now.

Reviewed By: beauby

Differential Revision: D27231000

fbshipit-source-id: 474cc1b23118703ecb47ed4b8e0c5b000572eae8
This commit is contained in:
Luca Wehrstedt 2021-03-23 01:32:11 -07:00 committed by Facebook GitHub Bot
parent c22fc448cd
commit 8518b0ee55
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
third_party/tensorpipe.BUILD vendored
View File

@ -78,6 +78,7 @@ header_template_rule(
"#cmakedefine01 TENSORPIPE_HAS_SHM_TRANSPORT": "",
"#cmakedefine01 TENSORPIPE_HAS_CMA_CHANNEL": "",
"#cmakedefine01 TENSORPIPE_HAS_CUDA_IPC_CHANNEL": "",
"#cmakedefine01 TENSORPIPE_HAS_CUDA_GDR_CHANNEL": "",
"#cmakedefine01 TENSORPIPE_HAS_IBV_TRANSPORT": "",
"#cmakedefine01 TENSORPIPE_SUPPORTS_CUDA": "",
},