From 1b6efeab829f3eeda5b39bd47624bb60b3531b88 Mon Sep 17 00:00:00 2001 From: Georgi Gerganov Date: Wed, 5 Jul 2023 20:20:05 +0300 Subject: [PATCH] tests : fix test-grad0 --- scripts/sync-ggml.sh | 5 ++++- tests/test-grad0.c | 2 +- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/scripts/sync-ggml.sh b/scripts/sync-ggml.sh index 574e5180b..02ea6ec15 100755 --- a/scripts/sync-ggml.sh +++ b/scripts/sync-ggml.sh @@ -8,4 +8,7 @@ cp -rpv ../ggml/src/ggml-opencl.cpp ./ggml-opencl.cpp cp -rpv ../ggml/src/ggml-metal.h ./ggml-metal.h cp -rpv ../ggml/src/ggml-metal.m ./ggml-metal.m cp -rpv ../ggml/src/ggml-metal.metal ./ggml-metal.metal -cp -rpv ../ggml/include/ggml/ggml.h ./ggml.h +cp -rpv ../ggml/include/ggml/ggml.h ./ggml.h + +cp -rpv ../ggml/tests/test-opt.c ./tests/test-opt.c +cp -rpv ../ggml/tests/test-grad0.c ./tests/test-grad0.c diff --git a/tests/test-grad0.c b/tests/test-grad0.c index b5a499c1d..a3e25214b 100644 --- a/tests/test-grad0.c +++ b/tests/test-grad0.c @@ -1154,7 +1154,7 @@ int main(int argc, const char ** argv) { continue; } - struct ggml_tensor * f = ggml_sum(ctx0, ggml_rope(ctx0, x[0], n_past, n_rot, mode)); + struct ggml_tensor * f = ggml_sum(ctx0, ggml_rope(ctx0, x[0], n_past, n_rot, mode, 0)); GGML_PRINT_DEBUG("rope: n_past: %d n_rot: %d mode: %d\n", n_past, n_rot, mode); check_gradient("rope", ctx0, x, f, ndims, nargs, 1e-2f, 1e-3f, INFINITY);