format

microsoft · Oct 5, 2024 · f8f2115 · f8f2115
1 parent edae0c4
commit f8f2115
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -3,8 +3,8 @@
 # Licensed under the MIT License.
 # --------------------------------------------------------------------------
 
-from pathlib import Path
 import os
+from pathlib import Path
 
 from setuptools import find_packages, setup
 from torch.utils.cpp_extension import BuildExtension, CUDAExtension
@@ -28,7 +28,7 @@ def build_cuda_extensions():
         print("TORCH_CUDA_ARCH_LIST is not set, compiling for all arch")
     else:
         TORCH_CUDA_ARCH_LIST = TORCH_CUDA_ARCH_LIST.split(' ')
-        compute_capabilities = [int(10*float(arch)) for arch in TORCH_CUDA_ARCH_LIST]
+        compute_capabilities = [int(10 * float(arch)) for arch in TORCH_CUDA_ARCH_LIST]
     for cap in compute_capabilities:
         arch_flags += ["-gencode", f"arch=compute_{cap},code=sm_{cap}"]
     extra_compile_args = {