eliaskohout/ctensor
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fixed memory leak.

Browse Source
This commit is contained in:
Elias Kohout
2023-09-07 13:48:56 +02:00
parent 0af2024991
commit 3a4ecebcfa
1 changed files with 24 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
tensor.c
View File

@@ -361,8 +361,14 @@ tensor tensor_add(const tensor t1, const tensor t2)
tensor t3 = tensor_new(); tensor t3 = tensor_new();
if(t3 == NULL) return NULL; if(t3 == NULL) return NULL;
if (!tensor_cpy(t3, t1)) return NULL; if (!tensor_cpy(t3, t1)) {
if (!tensor_add_inplace(t3, t2)) return NULL; tensor_destroy(t3);
return NULL;
}
if (!tensor_add_inplace(t3, t2)) {
tensor_destroy(t3);
return NULL;
}
return t3; return t3;
} }
@@ -381,8 +387,14 @@ tensor tensor_sub(const tensor t1, const tensor t2)
tensor t3 = tensor_new(); tensor t3 = tensor_new();
if(t3 == NULL) return NULL; if(t3 == NULL) return NULL;
if (!tensor_cpy(t3, t1)) return NULL; if (!tensor_cpy(t3, t1)) {
if (!tensor_sub_inplace(t3, t2)) return NULL; tensor_destroy(t3);
return NULL;
}
if (!tensor_sub_inplace(t3, t2)) {
tensor_destroy(t3);
return NULL;
}
return t3; return t3;
} }
@@ -402,8 +414,14 @@ tensor tensor_mul(const tensor t1, const tensor t2)
tensor t3 = tensor_new(); tensor t3 = tensor_new();
if(t3 == NULL) return NULL; if(t3 == NULL) return NULL;
if (!tensor_cpy(t3, t1)) return NULL; if (!tensor_cpy(t3, t1)) {
if (!tensor_mul_inplace(t3, t2)) return NULL; tensor_destroy(t3);
return NULL;
}
if (!tensor_mul_inplace(t3, t2)) {
tensor_destroy(t3);
return NULL;
}
return t3; return t3;
} }