Unverified Commit 19fa0e57 authored by Yasyf Mohamedali's avatar Yasyf Mohamedali Committed by GitHub
Browse files

Remove extraneous comma (#2993)

Prevents `TypeError: category must be a Warning subclass, not 'str'`.
parent 3a5d93bc
...@@ -157,7 +157,7 @@ def set_cuda_arch_list(cuda_dir): ...@@ -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'
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment