fix code style

open-mmlab · May 11, 2023 · 9f4fb86 · 9f4fb86
1 parent 16c1def
commit 9f4fb86
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -389,9 +389,11 @@ def get_mluops_version(file_path):
                     './mlu-ops/bangc-ops/kernels/**/*.cpp', recursive=True) + \
                 glob.glob(
                     './mlu-ops/bangc-ops/kernels/**/*.mlu', recursive=True)
-            extra_link_args=['-Wl,--whole-archive',
+            extra_link_args = [
+                '-Wl,--whole-archive',
                 './mlu-ops/bangc-ops/kernels/kernel_wrapper/lib/libextops.a',
-                '-Wl,--no-whole-archive']
+                '-Wl,--no-whole-archive'
+            ]
             extension = MLUExtension
             include_dirs.append(os.path.abspath('./mmcv/ops/csrc/common'))
             include_dirs.append(os.path.abspath('./mmcv/ops/csrc/common/mlu'))