improve performance of sum backward pass

use add1(x,y) instead of add(x,repeat(y,x))
This commit is contained in:
xaedes 2023-04-26 00:43:02 +02:00
parent 0197bcb0ff
commit bfe507213c
No known key found for this signature in database
GPG key ID: 30030EDD817EA2B1

4
ggml.c
View file

@ -12640,9 +12640,9 @@ static void ggml_compute_backward(struct ggml_context * ctx, struct ggml_tensor
{
if (src0->grad) {
src0->grad =
ggml_add_impl(ctx,
ggml_add1_impl(ctx,
src0->grad,
ggml_repeat(ctx, tensor->grad, src0->grad),
tensor->grad,
inplace);
}
} break;