Add computation of #parameters, #MACC, and max feature map size in the tests

Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-11-23 13:07:54 +01:00
parent 04b4a69107
commit a17e7800b9
7 changed files with 58 additions and 1 deletions
@@ -166,6 +166,11 @@ Conv2d::Conv2d( Network *net, int out_ch, int kernelH, int kernelW,
    }
    initCUDNN(deConv);

+    if(this->groups != 1)
+        MACC = kernelH*kernelW*output_dim.c*output_dim.w*output_dim.h;
+    else
+        MACC = input_dim.c*kernelH*kernelW*output_dim.c*output_dim.w*output_dim.h;
+
    // allocate warkspace
    if (ws_sizeInBytes!=0) {
        checkCuda( cudaMalloc(&workSpace, ws_sizeInBytes) );