[Style] Fix lint

Author: tridao

pyproject.toml

@@ -42,6 +42,7 @@ torch = [
 
 [tool.ruff]
 line-length = 100
+exclude = [".#*"]
 
 [tool.ruff.lint]
 ignore = [

quack/gemm.py

@@ -170,13 +170,13 @@ def gemm(
     device_capacity = get_device_capacity(A.device)
     assert device_capacity[0] in [9, 10, 11], "Only SM90, SM100, and SM110 are supported"
     if rounding_mode == RoundingMode.RS:
-        assert (
-            device_capacity[0] >= 10
-        ), "Stochastic rounding (RoundingMode.RS) requires SM100+ (Blackwell)"
+        assert device_capacity[0] >= 10, (
+            "Stochastic rounding (RoundingMode.RS) requires SM100+ (Blackwell)"
+        )
     if is_dynamic_persistent and device_capacity[0] == 9:
-        assert (
-            tile_count_semaphore is not None
-        ), "Dynamic persistent tile scheduler in SM90 requires a semaphore in GMEM"
+        assert tile_count_semaphore is not None, (
+            "Dynamic persistent tile scheduler in SM90 requires a semaphore in GMEM"
+        )
 
     A_p, B_p, D_p, C_p = perm3d(A, B, D, C, varlen_m=varlen_m, varlen_k=varlen_k)
     a_major, b_major, d_major, c_major = get_majors(A_p, B_p, D_p, C_p)

quack/gemm_act.py

@@ -187,15 +187,15 @@ class GemmGatedMixin(GemmActMixin):
     def epi_to_underlying_arguments(
         self, args: GemmActMixin.EpilogueArguments, *, loc=None, ip=None
     ) -> GemmActMixin.EpilogueParams:
-        assert (
-            args.mPostAct.element_type.width == 16
-        ), "GemmGated only supports 16bit postact for now"
+        assert args.mPostAct.element_type.width == 16, (
+            "GemmGated only supports 16bit postact for now"
+        )
         assert self.d_layout is None or self.d_layout.is_n_major_c()
         assert cutlass.utils.LayoutEnum.from_tensor(args.mPostAct).is_n_major_c()
         if self.arch == 90:
-            assert (
-                self.cta_tile_shape_mnk[1] % 32 == 0
-            ), "GemmGatedSm90 requires tileN to be divisible by 32"
+            assert self.cta_tile_shape_mnk[1] % 32 == 0, (
+                "GemmGatedSm90 requires tileN to be divisible by 32"
+            )
         self.rounding_mode = args.rounding_mode
         self.postact_dtype = args.mPostAct.element_type
         self.postact_layout = cutlass.utils.LayoutEnum.from_tensor(args.mPostAct)
@@ -415,14 +415,14 @@ def gemm_act(
     device_capacity = get_device_capacity(A.device)
     assert device_capacity[0] in [9, 10, 11], "Only SM90, SM100, and SM110 are supported"
     if rounding_mode == RoundingMode.RS:
-        assert (
-            device_capacity[0] >= 10
-        ), "Stochastic rounding (RoundingMode.RS) requires SM100+ (Blackwell)"
+        assert device_capacity[0] >= 10, (
+            "Stochastic rounding (RoundingMode.RS) requires SM100+ (Blackwell)"
+        )
 
     if is_dynamic_persistent and device_capacity[0] == 9:
-        assert (
-            tile_count_semaphore is not None
-        ), "Dynamic persistent tile scheduler in SM90 requires a semaphore in GMEM"
+        assert tile_count_semaphore is not None, (
+            "Dynamic persistent tile scheduler in SM90 requires a semaphore in GMEM"
+        )
 
     sr_seed_mode = (
         2 if isinstance(sr_seed, Tensor) else (1 if rounding_mode == RoundingMode.RS else 0)

quack/gemm_dact.py

@@ -416,9 +416,9 @@ def gemm_dact(
     assert device_capacity[0] in [9, 10, 11], "Only SM90, SM100, and SM110 are supported"
 
     if is_dynamic_persistent and device_capacity[0] == 9:
-        assert (
-            tile_count_semaphore is not None
-        ), "Dynamic persistent tile scheduler in SM90 requires a semaphore in GMEM"
+        assert tile_count_semaphore is not None, (
+            "Dynamic persistent tile scheduler in SM90 requires a semaphore in GMEM"
+        )
 
     compiled_fn = _compile_gemm_dact(
         a_dtype,

quack/gemm_norm_act.py

@@ -312,9 +312,9 @@ def gemm_norm_act_fn(
         assert device_capacity[0] >= 10, "Stochastic rounding requires SM100+"
 
     if is_dynamic_persistent and device_capacity[0] == 9:
-        assert (
-            tile_count_semaphore is not None
-        ), "Dynamic persistent tile scheduler in SM90 requires a semaphore in GMEM"
+        assert tile_count_semaphore is not None, (
+            "Dynamic persistent tile scheduler in SM90 requires a semaphore in GMEM"
+        )
 
     sr_seed_mode = (
         2 if isinstance(sr_seed, Tensor) else (1 if rounding_mode == RoundingMode.RS else 0)

quack/gemm_sq_reduce.py

@@ -199,9 +199,9 @@ def gemm_sq_reduce(
     a_dtype, b_dtype, d_dtype, c_dtype = get_dtypes(A, B, D, C)
 
     if is_dynamic_persistent and device_capacity[0] == 9:
-        assert (
-            tile_count_semaphore is not None
-        ), "Dynamic persistent tile scheduler in SM90 requires a semaphore in GMEM"
+        assert tile_count_semaphore is not None, (
+            "Dynamic persistent tile scheduler in SM90 requires a semaphore in GMEM"
+        )
 
     compiled_fn = _compile_gemm_sq_reduce(
         a_dtype,

quack/gemm_symmetric.py

@@ -317,9 +317,9 @@ def gemm_symmetric(
     assert device_capacity[0] in [9, 10, 11], "Only SM90, SM100, and SM110 are supported"
 
     if is_dynamic_persistent and device_capacity[0] == 9:
-        assert (
-            tile_count_semaphore is not None
-        ), "Dynamic persistent tile scheduler in SM90 requires a semaphore in GMEM"
+        assert tile_count_semaphore is not None, (
+            "Dynamic persistent tile scheduler in SM90 requires a semaphore in GMEM"
+        )
 
     tile_shape_mn = (tile_M, tile_N)
     cluster_shape_mnk = (cluster_M, cluster_N, 1)