Commit Graph

119 Commits

Author SHA1 Message Date
591507ed21 Split tests to cpu and cuda 2024-09-07 21:24:34 +02:00
f8220f0ec1 Restructure cuda backend 2024-09-05 22:23:47 +02:00
65727dfee8 Move cuda source to separate directory 2024-09-05 17:39:09 +02:00
bc9bff10cd Load running mean and var from weight file 2024-08-25 19:33:33 +02:00
9704d0d53e Add running mean and running var to batchnorm 2024-08-25 19:05:10 +02:00
a4a2801ee4 Fix missing activation in batch norm 2024-06-02 14:41:14 +02:00
8168f02f58 Add adaptive avg pooling 2024-05-30 17:17:31 +02:00
479c1119e7 Fix module layer issues 2024-05-30 13:08:13 +02:00
8ac2da004c Fix adding module layers 2024-05-29 21:02:26 +02:00
046c1b0cc2 Improve model summary 2024-05-28 19:24:56 +02:00
098fd65074 Add model summary 2024-05-28 18:42:58 +02:00
df47a31f36 Rename dim2d to shape2d 2024-05-27 21:14:51 +02:00
94a16b4352 Add padding to max pooling 2024-05-26 19:03:10 +02:00
4a67b708f0 Add padding to avg pooling 2024-05-26 18:54:12 +02:00
cba177e417 Add getOutputDims to 2d layers 2024-05-26 14:28:43 +02:00
37c29d6734 Fix wrong conv2d output size calculation 2024-05-23 21:38:11 +02:00
6dca8ccd3c Unify 2d layer naming 2024-05-20 16:23:58 +02:00
74098b24e3 Add support for non square matrices 2024-05-20 15:20:43 +02:00
6f8b5f4081 Rename batchnorm 2024-05-20 13:05:48 +02:00
b5fb205df8 Update module add layer 2024-05-19 20:22:08 +02:00
33d4a43dca Update concat layer 2024-05-19 20:21:13 +02:00
e23ffe1ee1 Remove default constructor from Module 2024-05-19 16:16:57 +02:00
4a1c4a5f91 Add epsilon param to batch norm 2024-05-19 15:13:22 +02:00
c84f58b97c Implement modules 2024-05-15 18:52:31 +02:00
7c48ed86d2 Implement vector variance function 2024-05-14 21:58:23 +02:00
33a3c6afcb Fix batch norm layer 2024-05-14 21:42:01 +02:00
da8f3167cb Add utils vector mean function 2024-05-14 21:41:18 +02:00
5c8d3f7e25 Compute mean and variance 2024-04-29 20:55:11 +02:00
0ab623fa23 Implement vector mean calculation 2024-04-28 22:04:15 +02:00
96536a44fd Implement batch norm test 2024-04-28 21:46:42 +02:00
f60d62f6bd Implement batch norm layer 2024-04-28 19:58:00 +02:00
1f708f3ea3 Change vector scalar math kernels 2024-04-28 19:03:06 +02:00
69111f6cb1 Add version to model bin format 2024-04-23 19:53:48 +02:00
74262309ad Fix max pooling output size calc 2024-04-22 20:58:02 +02:00
c2acad151b Implement simple model validation 2024-04-22 20:57:40 +02:00
f17debc244 Implement getOutputSize and getInputSize for seq layers 2024-04-22 20:31:58 +02:00
a32c737785 Allocate activation on heap 2024-04-22 18:59:16 +02:00
26cea9b12c Fix more cuda memcheck errors 2024-04-21 22:56:55 +02:00
58af95eb25 Remove cudaDeviceReset calls from tests 2024-04-21 22:47:12 +02:00
bdbb3f978e Fix matmul and max reduce memcheck errors 2024-04-21 22:11:02 +02:00
9a6152469a Update activation test 2024-04-21 14:00:43 +02:00
942ee6a32b Add layer name to vector 2024-04-21 12:20:02 +02:00
0170afaf3f Improve cuda error handling 2024-04-21 12:19:19 +02:00
d64a28bc9c Fix model weights export 2024-04-21 00:05:56 +02:00
9c5d853b75 Fix bin file seek offset 2024-04-20 21:30:01 +02:00
5e663b9029 Fix bias in conv layer 2024-04-20 19:09:00 +02:00
d08567a563 Fix weigh bias parsing and better error logging 2024-04-20 18:36:53 +02:00
ecf7416f8e Rework padding size setting 2024-04-20 16:31:28 +02:00
432adf57bd Test model weights loading 2024-04-16 21:07:06 +02:00
9fb9d7e8e1 Implement getting layer, weights and biases 2024-04-16 19:09:41 +02:00