Skip to content

Commit bc19808

Browse files
committed
Fix format
Signed-off-by: Hongwei <hongweichen@microsoft.com>
1 parent fa70484 commit bc19808

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

op_builder/builder.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,7 @@ def get_default_compute_capabilities():
6363
compute_caps = DEFAULT_COMPUTE_CAPABILITIES
6464
# Update compute capability according to: https://en.wikipedia.org/wiki/CUDA#GPUs_supported
6565
import torch.utils.cpp_extension
66-
if torch.utils.cpp_extension.CUDA_HOME is not None:
66+
if torch.utils.cpp_extension.CUDA_HOME is not None:
6767
if installed_cuda_version()[0] == 11:
6868
if installed_cuda_version()[1] >= 0:
6969
compute_caps += ";8.0"

0 commit comments

Comments
 (0)