fmt

Author: oraluben

docs/conf.py

@@ -1,4 +1,3 @@
-
 # General information about the project.
 project = "Tile Language <br>"
 author = "Tile Lang Contributors"

examples/blocksparse_attention/example_tilelang_sparse_gqa_decode_varlen_indice.py

@@ -369,7 +369,9 @@ def debug(name, expect, actual, atol=1e-3, rtol=1e-3):
     print(name + f"  all_close={all_close}")
     if not all_close:
         diff = (expect - actual).abs()
-        print(f"all_close={all_close}, max={diff.max().item()}, min={diff.min().item()}, mean={diff.mean().item()}")
+        print(
+            f"all_close={all_close}, max={diff.max().item()}, min={diff.min().item()}, mean={diff.mean().item()}"
+        )
         max_indices = torch.nonzero(diff == diff.max().item())
         first_index = tuple(max_indices[0].tolist())
         print(f"Index: {first_index}, expect: {expect[first_index]}, actual: {actual[first_index]}")

examples/blocksparse_attention/example_tilelang_sparse_gqa_decode_varlen_mask.py

@@ -359,7 +359,9 @@ def debug(name, expect, actual, atol=1e-3, rtol=1e-3):
         # print(expect[3, 28])
         # print(actual[3, 28])
         diff = (expect - actual).abs()
-        print(f"all_close={all_close}, max={diff.max().item()}, min={diff.min().item()}, mean={diff.mean().item()}")
+        print(
+            f"all_close={all_close}, max={diff.max().item()}, min={diff.min().item()}, mean={diff.mean().item()}"
+        )
         max_indices = torch.nonzero(diff == diff.max().item())
         first_index = tuple(max_indices[0].tolist())
         print(f"Index: {first_index}, expect: {expect[first_index]}, actual: {actual[first_index]}")