fix typo

tgale96 · Jan 11, 2024 · 5ef3699 · 5ef3699
1 parent 37be1fb
commit 5ef3699
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -18,10 +18,10 @@
 ]
 
 if device_capability:
-    nvcc_flags.extend(
+    nvcc_flags.extend([
         f"--generate-code=arch=compute_{device_capability},code=sm_{device_capability}",
         f"-DGROUPED_GEMM_DEVICE_CAPABILITY={device_capability}",
-    )
+    ])
 
 ext_modules = [
     CUDAExtension(