1 files changed, 4 insertions, 1 deletions
diff --git a/src/mtx_mul.c b/src/mtx_mul.c
index ef0f12b..8272d53 100644
--- a/src/mtx_mul.c
+++ b/src/mtx_mul.c
@@ -24,13 +24,16 @@
 */
 
 t_matrixfloat*mtx_doMultiply(int rowA, t_matrixfloat*A, int colArowB, t_matrixfloat*B, int colB){
-  t_matrixfloat*result=(t_matrixfloat*)getbytes(sizeof(t_matrixfloat)*rowA*colB);
+  t_matrixfloat*result=0;
+  if(!A || !B || !rowA || !colArowB || !colB)return 0;
+  result=(t_matrixfloat*)getbytes(sizeof(t_matrixfloat)*rowA*colB);
   int r, c, n;
   for(r=0; r<rowA; r++){
     for(c=0; c<colB; c++){
       t_matrixfloat sum=0.f;
       for(n=0;n<colArowB; n++)
         sum+=A[colArowB*r+n]*B[colB*n+c];
+      result[colB*r+c]=sum;
     }
   }
   return result;