Browse Source

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
commit
19fa0e57f6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 2
      op_builder/utils.py

2
op_builder/utils.py

@ -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
if not cuda_available:
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'
'By default, Colossal-AI will cross-compile for \n'
'1. Pascal (compute capabilities 6.0, 6.1, 6.2),\n'

Loading…
Cancel
Save