(x, qw, s, b, i1, i2)
| 932 | return mx.gather_mm(x, w, i1, i2).sum() |
| 933 | |
| 934 | def f_test(x, qw, s, b, i1, i2): |
| 935 | return mx.gather_qmm(x, qw, s, b, i1, i2, transpose=True).sum() |
| 936 | |
| 937 | r1 = f_ref(x, w_hat, lhs_indices, rhs_indices) |
| 938 | r2 = f_test(x, qw, s, b, lhs_indices, rhs_indices) |
nothing calls this directly
no test coverage detected