metal : extend l2_norm support for non-cont src0 (#19502)

This commit is contained in:
Georgi Gerganov
2026-02-11 14:53:19 +02:00
committed by GitHub
parent ada90bf2ba
commit 9ab072ebbe
5 changed files with 75 additions and 30 deletions
+14 -1
View File
@@ -539,8 +539,21 @@ typedef struct {
typedef struct {
int32_t ne00;
int32_t ne00_4;
int32_t ne01;
int32_t ne02;
int32_t ne03;
uint64_t nb00;
uint64_t nb01;
uint64_t nb02;
uint64_t nb03;
int32_t ne0;
int32_t ne1;
int32_t ne2;
int32_t ne3;
uint64_t nb0;
uint64_t nb1;
uint64_t nb2;
uint64_t nb3;
float eps;
} ggml_metal_kargs_l2_norm;