Update ggml_extend.hpp

idostyle · web-flow · commit ecd661753a8d · 2025-01-22T13:08:20.000+01:00
diff --git a/ggml_extend.hpp b/ggml_extend.hpp
@@ -1310,7 +1310,7 @@ class Linear : public UnaryBlock {
         }
 
         auto im = ggml_mul_mat(ctx, x, w);
-        auto imv = ggml_diag(im);
+        auto imv = ggml_diag(ctx, im);
         
         // 
         print_ggml_tensor(x, true, "imatrix x");

Original file line number	Diff line number	Diff line change
`@@ -1310,7 +1310,7 @@ class Linear : public UnaryBlock {`
`1310`	`1310`	`}`
`1311`	`1311`
`1312`	`1312`	`auto im = ggml_mul_mat(ctx, x, w);`
`1313`		`- auto imv = ggml_diag(im);`
	`1313`	`+ auto imv = ggml_diag(ctx, im);`
`1314`	`1314`
`1315`	`1315`	`//`
`1316`	`1316`	`print_ggml_tensor(x, true, "imatrix x");`