zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-12-16 01:49:05 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Rename block sparse (#1149)

Browse Source 
* block_sparse_mm to gather_mm

* rename

* nit

* nit
This commit is contained in:
Awni Hannun
2024-05-22 07:48:34 -07:00
committed by GitHub
parent e6fecbb3e1
commit d568c7ee36
16 changed files with 120 additions and 111 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
mlx/ops.h
View File

@@ -1158,7 +1158,7 @@ array dequantize(
StreamOrDevice s = {});
/** Compute matrix products with matrix-level gather. */
array block_sparse_qmm(
array gather_qmm(
const array& x,
const array& w,
const array& scales,
@@ -1210,7 +1210,7 @@ array block_masked_mm(
StreamOrDevice s = {});
/** Compute matrix product with matrix-level gather */
array block_sparse_mm(
array gather_mm(
array a,
array b,
std::optional<array> lhs_indices = std::nullopt,