From 0fb5433e97cbdd27485c64dc47b470ae1f999c39 Mon Sep 17 00:00:00 2001 From: Howard Su Date: Sat, 1 Apr 2023 15:09:06 +0800 Subject: [PATCH] miss some changes in the previous commit --- ggml.c | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/ggml.c b/ggml.c index 3a225eaf1..590870056 100644 --- a/ggml.c +++ b/ggml.c @@ -4618,14 +4618,14 @@ struct ggml_tensor * ggml_diag_mask_inf( struct ggml_context * ctx, struct ggml_tensor * a, int n_past) { - ggml_diag_mask_inf_impl(ctx, a, n_past, false); + return ggml_diag_mask_inf_impl(ctx, a, n_past, false); } struct ggml_tensor * ggml_diag_mask_inf_inplace( struct ggml_context * ctx, struct ggml_tensor * a, int n_past) { - ggml_diag_mask_inf_impl(ctx, a, n_past, true); + return ggml_diag_mask_inf_impl(ctx, a, n_past, true); } // ggml_soft_max @@ -4654,12 +4654,12 @@ struct ggml_tensor * ggml_soft_max_impl( struct ggml_tensor * ggml_soft_max( struct ggml_context * ctx, struct ggml_tensor * a) { - ggml_soft_max_impl(ctx, a, false); + return ggml_soft_max_impl(ctx, a, false); } struct ggml_tensor * ggml_soft_max_inplace( struct ggml_context * ctx, struct ggml_tensor * a) { - ggml_soft_max_impl(ctx, a, true); + return ggml_soft_max_impl(ctx, a, true); } // ggml_rope @@ -4699,7 +4699,7 @@ struct ggml_tensor * ggml_rope( int n_past, int n_dims, int mode){ - ggml_rope_impl(ctx, a, n_past, n_dims, mode, false); + return ggml_rope_impl(ctx, a, n_past, n_dims, mode, false); } struct ggml_tensor * ggml_rope_inplace( @@ -4708,7 +4708,7 @@ struct ggml_tensor * ggml_rope_inplace( int n_past, int n_dims, int mode){ - ggml_rope_impl(ctx, a, n_past, n_dims, mode, true); + return ggml_rope_impl(ctx, a, n_past, n_dims, mode, true); } // ggml_conv_1d_1s