Defined totalElements

nlesc-recruit · Dec 13, 2024 · 573f35e · 573f35e
1 parent 9fdee0e
commit 573f35e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/matlab/beamform_mex.cu b/matlab/beamform_mex.cu
@@ -34,7 +34,7 @@ void mexFunction(int nlhs, mxArray *plhs[], int nrhs, const mxArray *prhs[]) {
   int32_t *outData = static_cast<int32_t *>(mxGetData(outArray));
 
   // Copy the data
-  std::memcpy(outData, BF, totalElements * sizeof(int32_t));
+  std::memcpy(outData, BF, 2 * frames * samples * sizeof(int32_t));
 
   // Assign output
   plhs[0] = outArray;