github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Remove extraneous comma (#2993) 

Prevents `TypeError: category must be a Warning subclass, not 'str'`.
pull/2994/head
Yasyf Mohamedali 2023-03-03 22:44:06 -08:00 committed by GitHub
parent 3a5d93bc2c
commit 19fa0e57f6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
op_builder/utils.py
View File

@ -157,7 +157,7 @@ def set_cuda_arch_list(cuda_dir):
# we only need to set this when CUDA is not available for cross-compilation # we only need to set this when CUDA is not available for cross-compilation
if not cuda_available: if not cuda_available:
warnings.warn( warnings.warn(
'\n[extension] PyTorch did not find available GPUs on this system.\n', '\n[extension] PyTorch did not find available GPUs on this system.\n'
'If your intention is to cross-compile, this is not an error.\n' 'If your intention is to cross-compile, this is not an error.\n'
'By default, Colossal-AI will cross-compile for \n' 'By default, Colossal-AI will cross-compile for \n'
'1. Pascal (compute capabilities 6.0, 6.1, 6.2),\n' '1. Pascal (compute capabilities 6.0, 6.1, 6.2),\n'