Skip to content

Commit 94e55e1

Browse files
committed
fix some tle-dist bugs
1 parent 389077b commit 94e55e1

2 files changed

Lines changed: 0 additions & 2 deletions

File tree

python/test/tle/unit/test_tle_distributed_d2d.sh

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,6 @@ export FLAGCX_USE_HETERO_COMM=1
77
export FLAGCX_MEM_ENABLE=1
88
export FLAGCX_VMM_ENABLE=0
99
export FLAGCX_P2P_DISABLE=1
10-
export CUDA_VISIBLE_DEVICES=0,1
1110

1211
port=8333
1312
# Check whether port is occupied

python/test/tle/unit/test_tle_get_local_pe.sh

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,6 @@ export FLAGCX_USE_HETERO_COMM=1
1717
export FLAGCX_MEM_ENABLE=1
1818
export FLAGCX_VMM_ENABLE=0
1919
export FLAGCX_P2P_DISABLE=1
20-
export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7
2120
# Need to preload customized gloo library specified for FlagCX linkage
2221
#export LD_PRELOAD=/usr/local/lib/libgloo.so
2322
#export LD_PRELOAD=/usr/local/nccl/build/lib/libnccl.so

0 commit comments

Comments
 (0)