Skip to content

Commit 0115df2

Browse files
authored
metal : small-batch mat-mul kernels (#10581)
* metal : small-batch mat-mul kernels ggml-ci * metal : add rest of types ggml-ci * metal : final adjustments ggml-ci * metal : add comments ggml-ci
1 parent 515d4e5 commit 0115df2

File tree

4 files changed

+709
-29
lines changed

4 files changed

+709
-29
lines changed

ggml/src/ggml-metal/ggml-metal-impl.h

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -192,6 +192,30 @@ typedef struct {
192192
int16_t r3;
193193
} ggml_metal_kargs_mul_mv;
194194

195+
typedef struct {
196+
int32_t ne00;
197+
int32_t ne01;
198+
int32_t ne02;
199+
uint64_t nb00;
200+
uint64_t nb01;
201+
uint64_t nb02;
202+
uint64_t nb03;
203+
int32_t ne10;
204+
int32_t ne11;
205+
int32_t ne12;
206+
uint64_t nb10;
207+
uint64_t nb11;
208+
uint64_t nb12;
209+
uint64_t nb13;
210+
int32_t ne0;
211+
int32_t ne1;
212+
int16_t r2;
213+
int16_t r3;
214+
int16_t nsg;
215+
int16_t nxpsg;
216+
int16_t r1ptg;
217+
} ggml_metal_kargs_mul_mv_ext;
218+
195219
typedef struct {
196220
int32_t nei0;
197221
int32_t nei1;

0 commit comments

Comments
 (0)