Block sparse mm (#1058)

2025-12-14 00:39:06 +08:00 · 2024-05-02 14:03:58 -07:00
parent 17f57df797
commit f390957685
15 changed files with 1323 additions and 75 deletions
--- a/docs/src/python/ops.rst
+++ b/docs/src/python/ops.rst
@@ -32,6 +32,7 @@ Operations
   bitwise_or
   bitwise_xor
   block_masked_mm
+   block_sparse_mm
   broadcast_to
   ceil
   clip