File tree Expand file tree Collapse file tree 1 file changed +5
-5
lines changed Expand file tree Collapse file tree 1 file changed +5
-5
lines changed Original file line number Diff line number Diff line change 33]
44
55import os
6- from importlib import import_module , util
6+ from importlib import util
77from typing import Optional
88
99
@@ -23,10 +23,10 @@ def nccl_import(message: Optional[str] = None) -> str:
2323 else :
2424 # if unable to import but the package is installed
2525 nccl_message = (
26- f "cupy is installed but cupy.cuda.nccl not available, Falling back to pure MPI."
27- "Please ensure your CUDA NCCL environment is set up correctly "
28- "for more details visit 'https://docs.cupy.dev/en/stable/install.html'"
29- )
26+ "cupy is installed but cupy.cuda.nccl not available, Falling back to pure MPI."
27+ "Please ensure your CUDA NCCL environment is set up correctly "
28+ "for more details visit 'https://docs.cupy.dev/en/stable/install.html'"
29+ )
3030 print (UserWarning (nccl_message ))
3131 else :
3232 nccl_message = (
You can’t perform that action at this time.
0 commit comments