free arrays only if allocated

milk-org · Sep 15, 2024 · fdba300 · fdba300
1 parent 4fa8677
commit fdba300
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/plugins/milk-extra-src/linalgebra/SGEMM.c b/plugins/milk-extra-src/linalgebra/SGEMM.c
@@ -646,14 +646,15 @@ errno_t computeSGEMM(
                     0.0, outimg->im->array.F, outMdim);
     }
 
+    printf("Freeing float arrays\n");
+    fflush(stdout);
 
-
-    if (imginA.md->datatype == _DATATYPE_FLOAT )
+    if (imginA.md->datatype != _DATATYPE_FLOAT )
     {
         free(imarrayA);
     }
 
-    if (imginB.md->datatype == _DATATYPE_FLOAT )
+    if (imginB.md->datatype != _DATATYPE_FLOAT )
     {
         free(imarrayB);
     }