vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

test-grad0.c add TODO for view_2d and view_3d

Browse source 
add_at (required for view backward pass) is a bit tricky for n_dims > 1.
This commit is contained in:
xaedes 2023-04-28 18:16:55 +02:00
parent c601df973c
commit 1997152f7f
No known key found for this signature in database
GPG key ID: 30030EDD817EA2B1
1 changed files with 1 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
tests/test-grad0.c
View file

@ -587,6 +587,7 @@ int main(int argc, const char ** argv) {
const int offset = i0 * sizeof(float);
const int nelem = i1 - i0;
// TODO : test for view_2d and view_3d
struct ggml_tensor * f = ggml_sum(ctx0, ggml_view_1d(ctx0, x[0], nelem, offset));
check_gradient("view", ctx0, x, f, ndims, nargs, 1e-3f, 1e-3f, INFINITY);