perf: remove unnecessary contiguous operation in block sparse attenti…

…on (#561) The contiguous operation is no longer required after #513
flashinfer-ai · Oct 26, 2024 · 7a7ad46 · 7a7ad46
1 parent 3fbf028
commit 7a7ad46
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/python/flashinfer/sparse.py b/python/flashinfer/sparse.py
@@ -435,9 +435,8 @@ def run(
             rope_scale = 1.0
         if rope_theta is None:
             rope_theta = 1e4
-
-        k = k.reshape(-1, self.C, *k.shape[-2:]).contiguous()
-        v = v.reshape(-1, self.C, *v.shape[-2:]).contiguous()
+        k = k.reshape(-1, self.C, *k.shape[-2:])
+        v = v.reshape(-1, self.C, *v.shape[-2:])
 
         lse = None
         if return_lse: