diff options
Diffstat (limited to 'tests/test-grad0.c')
| -rw-r--r-- | tests/test-grad0.c | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tests/test-grad0.c b/tests/test-grad0.c index ef20bce..6d31221 100644 --- a/tests/test-grad0.c +++ b/tests/test-grad0.c @@ -850,7 +850,7 @@ int main(int argc, const char ** argv) { ggml_set_param(ctx0, x[i]); } - struct ggml_tensor * f = ggml_sum(ctx0, ggml_rms_norm(ctx0, x[0])); + struct ggml_tensor * f = ggml_sum(ctx0, ggml_rms_norm(ctx0, x[0], 1e-6f)); check_gradient("rms_norm", ctx0, x, f, ndims, nargs, 1e-4f, 1.0f, INFINITY); } |