Skip to content

Commit 66d83ee

Browse files
authored
[CI] Remove nccl RAS workaround. (#11349)
1 parent dec7f58 commit 66d83ee

File tree

2 files changed

+0
-2
lines changed

2 files changed

+0
-2
lines changed

ops/docker_run.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,6 @@ def docker_run(
7070
docker_run_cli_args.extend(
7171
itertools.chain.from_iterable([["-e", f"{k}={v}"] for k, v in user_ids.items()])
7272
)
73-
docker_run_cli_args.extend(["-e", "NCCL_RAS_ENABLE=0"])
7473
docker_run_cli_args.extend(extra_args)
7574
docker_run_cli_args.append(image_uri)
7675
docker_run_cli_args.extend(command_args)

ops/pipeline/test-python-wheel-impl.sh

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,6 @@ case "$suite" in
4545
mgpu)
4646
echo "-- Run Python tests, using multiple GPUs"
4747
python -c 'from cupy.cuda import jitify; jitify._init_module()'
48-
export NCCL_RAS_ENABLE=0
4948
pytest -v -s -rxXs --durations=0 -m 'mgpu' tests/python-gpu
5049
pytest -v -s -rxXs --durations=0 tests/test_distributed/test_gpu_with_dask
5150
pytest -v -s -rxXs --durations=0 tests/test_distributed/test_gpu_with_spark

0 commit comments

Comments
 (0)