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

Bypass Clang-16 NCCL incompatibility.

Browse Source 
PiperOrigin-RevId: 515672644
This commit is contained in:
Juan Martinez Castellanos 2023-03-10 10:23:52 -08:00 committed by TensorFlower Gardener
parent 3abd29599a
commit c127e39cbb
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/dtensor/python/tests/BUILD
View File

@ -222,6 +222,7 @@ dtensor_test(
env = {
"DTENSOR_GPU_USE_NCCL_COMMUNICATION": "1",
"NCCL_P2P_DISABLE": "1", # FIXME(b/251183104): p2p detection in cuda 10.1+ is broken.
"NCCL_PROTO": "Simple,LL128", # FIXME(b/272050398): Delete this when the Clang-16/NCCL incompatibility has been resolved.
},
tags = [
"no_windows",
@ -266,6 +267,7 @@ dtensor_test(
env = {
"DTENSOR_GPU_USE_NCCL_COMMUNICATION": "1",
"NCCL_P2P_DISABLE": "1", # FIXME(b/251183104): p2p detection in cuda 10.1+ is broken.
"NCCL_PROTO": "Simple,LL128", # FIXME(b/272050398): Delete this when the Clang-16/NCCL incompatibility has been resolved.
},
tags = [
"no_windows",