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

test both gradients of mul_mat

Browse source
This commit is contained in:
xaedes 2023-04-24 22:37:09 +02:00
parent 20e3c1d2b4
commit 9345f4c3a5
No known key found for this signature in database
GPG key ID: 30030EDD817EA2B1
2 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
ggml.c
View file

@ -5819,7 +5819,7 @@ struct ggml_tensor * ggml_cont_impl(
bool is_node = false;
if (!inplace && a->grad) {
GGML_ASSERT(false); // TODO: implement backward
// TODO: implement backward
is_node = true;
}

3
tests/test-grad0.c
View file

@ -385,7 +385,7 @@ int main(int argc, const char ** argv) {
// mul_mat
{
const int nargs = 1;
const int nargs = 2;
for (int ndims = 2; ndims <= 2; ++ndims) {
x[0] = get_random_tensor(ctx0, ndims, ne, -1.0f, 1.0f);
@ -397,6 +397,7 @@ int main(int argc, const char ** argv) {
}
ggml_set_param(ctx0, x[0]);
ggml_set_param(ctx0, x[1]);
struct ggml_tensor * m = ggml_mul_mat(ctx0, x[1], x[0]);
struct ggml_tensor * f = ggml_sum(ctx0, m);