Masked mm (#978)

* Add block masked matmul op and primitive
2025-09-01 12:49:44 +08:00 · 2024-04-16 14:45:39 -07:00
parent 107ba2891a
commit b18468bf81
15 changed files with 1137 additions and 2 deletions
--- a/docs/src/python/ops.rst
+++ b/docs/src/python/ops.rst
@@ -29,6 +29,7 @@ Operations
   atleast_2d
   atleast_3d
   broadcast_to
+   block_masked_mm
   ceil
   clip
   concatenate