Skip to content

Commit 6b7df0b

Browse files
committed
Revert "fix check CUDA_DEVICE_MAX_CONNECTIONS"
This reverts commit 317f18c.
1 parent 317f18c commit 6b7df0b

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

internlm/utils/common.py

+2-3
Original file line numberDiff line numberDiff line change
@@ -250,9 +250,8 @@ def enable_pytorch_expandable_segments():
250250

251251

252252
def check_cuda_env():
253-
max_connections = os.getenv("CUDA_DEVICE_MAX_CONNECTIONS")
254-
assert max_connections is not None, "Env var CUDA_DEVICE_MAX_CONNECTIONS has not been set, please set it to 1!"
255-
assert max_connections == '1', "Env var CUDA_DEVICE_MAX_CONNECTIONS is set to {}, but it should be set to 1!".format(max_connections)
253+
if os.getenv("CUDA_DEVICE_MAX_CONNECTIONS") is None:
254+
logger.warning("Env var CUDA_DEVICE_MAX_CONNECTIONS has not be set, please note this!")
256255

257256

258257
class DummyProfile:

0 commit comments

Comments
 (0)