[SYCL][Matrix][Element-wise-ops] correct the reference multiplication

Signed-off-by: Dounia Khaldi <[email protected]>

Author: dkhaldi

SYCL/Matrix/element_wise_ops.cpp

@@ -95,10 +95,10 @@ void matrix_multiply(big_matrix<T1, NUM_ROWS_C, NUM_COLS_C> &C,
                                    sg_starty / SG_SZ * TN * 4,
                                N * 4, matrix_layout::packed_b);
              sub_c = joint_matrix_mad(sg, sub_a, sub_b, sub_c);
-             auto wi_slice_c = sub_c.get_wi_data();
-             for (int i = 0; i < wi_slice_c.length(); i++) {
-               wi_slice_c[i] *= 2;
-             }
+           }
+           auto wi_slice_c = sub_c.get_wi_data();
+           for (int i = 0; i < wi_slice_c.length(); i++) {
+             wi_slice_c[i] *= 2;
            }
            joint_matrix_store(sg, sub_c,
                               accC.get_pointer() + (sg_startx * TM) * N +
@@ -130,6 +130,7 @@ void matrix_multiply_ref(int32_t *A_mem, int32_t *B_mem, int32_t *C_mem, int M,
         }
         *(C_mem + m * N + n) = acc;
       }
+      *(C_mem + m * N + n) *= 2;
     }
 }