Ref : Added edge cases

jaywyawhare · jaywyawhare · commit cc18da125b44 · 2025-03-29T08:49:24.000+05:30
diff --git a/src/Layers/dense.c b/src/Layers/dense.c
@@ -40,6 +40,8 @@ int initialize_dense(DenseLayer *layer, int input_size, int output_size)
     if (layer->weights == (void *)CM_MEMORY_ALLOCATION_ERROR || layer->biases == (void *)CM_MEMORY_ALLOCATION_ERROR)
     {
         fprintf(stderr, "[initializeDense] Error: Memory allocation failed.\n");
+        cm_safe_free((void **)&layer->weights);
+        cm_safe_free((void **)&layer->biases);
         return CM_MEMORY_ALLOCATION_ERROR;
     }
 

Original file line number	Diff line number	Diff line change
`@@ -40,6 +40,8 @@ int initialize_dense(DenseLayer *layer, int input_size, int output_size)`
`40`	`40`	`if (layer->weights == (void )CM_MEMORY_ALLOCATION_ERROR \|\| layer->biases == (void )CM_MEMORY_ALLOCATION_ERROR)`
`41`	`41`	`{`
`42`	`42`	`fprintf(stderr, "[initializeDense] Error: Memory allocation failed.\n");`
	`43`	`+ cm_safe_free((void **)&layer->weights);`
	`44`	`+ cm_safe_free((void **)&layer->biases);`
`43`	`45`	`return CM_MEMORY_ALLOCATION_ERROR;`
`44`	`46`	`}`
`45`	`47`