Remove extraneous comma (#2993)

Prevents `TypeError: category must be a Warning subclass, not 'str'`.
pull/2994/head
Yasyf Mohamedali 2 years ago committed by GitHub
parent 3a5d93bc2c
commit 19fa0e57f6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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'

Loading…
Cancel
Save