(x, w, i1, i2)
| 929 | w_hat, qw, s, b = quantize(w) |
| 930 | |
| 931 | def f_ref(x, w, 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() |
nothing calls this directly
no outgoing calls
no test coverage detected