mirror of
https://github.com/ggml-org/llama.cpp.git
synced 2026-09-25 07:27:30 -05:00
add optional activation scales to mul_mat and mul_mat_id tensors
This commit is contained in:
+10
-2
@@ -3345,7 +3345,11 @@ struct ggml_tensor * ggml_mul_mat_ext(
|
||||
result->src[2] = scale_weight;
|
||||
}
|
||||
|
||||
GGML_UNUSED(scale_activations);
|
||||
if (scale_activations) {
|
||||
GGML_ASSERT(ggml_can_repeat(scale_activations, b));
|
||||
result->src[3] = scale_activations;
|
||||
}
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
@@ -3430,7 +3434,11 @@ struct ggml_tensor * ggml_mul_mat_id_ext(
|
||||
result->src[3] = s;
|
||||
}
|
||||
|
||||
GGML_UNUSED(scale_activations);
|
||||
if (scale_activations) {
|
||||
GGML_ASSERT(ggml_can_repeat(scale_activations, b));
|
||||
result->src[4] = scale_activations;
|
||||
}
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user