Renames the kernel generation script description to reflect sparse attention context

LoserCheems · LoserCheems · commit fc6414949270 · 2025-11-09T23:48:37.000+08:00
diff --git a/csrc/flash_sparse_attn/src/generate_kernels.py b/csrc/flash_sparse_attn/src/generate_kernels.py
@@ -113,7 +113,7 @@ def main(output_dir: Optional[str]) -> None:
 if __name__ == "__main__":
     parser = argparse.ArgumentParser(
         prog="generate_kernels",
-        description="Generate the flash_dmattn kernels template instantiations",
+        description="Generate the flash_sparse_attn kernels template instantiations",
     )
     parser.add_argument(
         "-o",

Original file line number	Diff line number	Diff line change
`@@ -113,7 +113,7 @@ def main(output_dir: Optional[str]) -> None:`
`113`	`113`	`if __name__ == "__main__":`
`114`	`114`	`parser = argparse.ArgumentParser(`
`115`	`115`	`prog="generate_kernels",`
`116`		`- description="Generate the flash_dmattn kernels template instantiations",`
	`116`	`+ description="Generate the flash_sparse_attn kernels template instantiations",`
`117`	`117`	`)`
`118`	`118`	`parser.add_argument(`
`119`	`119`	`"-o",`