tqcq/llama.cpp
0
0
Fork 0
mirror of https://github.com/ggml-org/llama.cpp.git synced 2025-06-27 03:55:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

special case gated ops

Browse Source
This commit is contained in:
Sigbjørn Skjæret
2025-06-13 01:07:49 +02:00
committed by Akarshan
parent 21c4963bd3
commit bb2fda70ae
1 changed files with 10 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
tests/test-backend-ops.cpp
View File

@ -1072,7 +1072,16 @@ struct test_unary : public test_case {
ggml_set_name(a, "a"); ggml_set_name(a, "a");
} }
ggml_tensor * out = ggml_unary(ctx, a, op); ggml_tensor * out;
if (op == GGML_UNARY_OP_REGLU) {
out = ggml_reglu(ctx, a);
} else if (op == GGML_UNARY_OP_GEGLU) {
out = ggml_geglu(ctx, a);
} else if (op == GGML_UNARY_OP_SWIGLU) {
out = ggml_swiglu(ctx, a);
} else {
out = ggml_unary(ctx, a, op);
}
ggml_set_name(out, "out"); ggml_set_name(out, "out");
return out; return out;