fix format

vectorch-ai · Feb 12, 2025 · 35ee21a · 35ee21a
1 parent fac9dcd
commit 35ee21a
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/kernels/attention/mla_kernel_sm80.cuh b/src/kernels/attention/mla_kernel_sm80.cuh
@@ -84,8 +84,10 @@ __global__ __launch_bounds__(Traits::kThreadNum) void mla_kernel_sm80(
 
   // Gmem
   // (BLK_M, BLK_K, STAGES)
-  Tensor gQ = local_tile(Q, Shape<_BLK_M, _BLK_K>{}, make_coord(m_block_idx, _));
-  Tensor gO = local_tile(O, Shape<_BLK_M, _BLK_K>{}, make_coord(m_block_idx, _));
+  Tensor gQ =
+      local_tile(Q, Shape<_BLK_M, _BLK_K>{}, make_coord(m_block_idx, _));
+  Tensor gO =
+      local_tile(O, Shape<_BLK_M, _BLK_K>{}, make_coord(m_block_idx, _));
   // (BLK_N, BLK_K, n, STAGES)
   Tensor gKV = local_tile(KV, Shape<_BLK_N, _BLK_K>{}, make_coord(_, _));