  *  @param[in] g_y All \f$P_{m,n}(Y)\f$
  *  @param[in] g_v All \f$P^T_{m,n}(V)\f$
  */
-__global__
-void algSAT_stage4( float *g_out,
+  __global__
+void algSAT_stage4(const int c_width, const int c_height, float *g_out,
     const float *g_in,
     const float *g_y,
     const float *g_v )

      g_ysum += c_m_size;
    }

        // fix ybar -> y -------------------------

    g_ybar += c_width;
    y = *g_ybar += y;

 *  @param[out] g_ysum All \f$s(P_{m,n}(Y))\f$
 */
__global__
void algSAT_stage2(const int c_n_size, const int c_m_size, const int c_width, float *g_ybar,
    float *g_ysum )
{
  const int tx = threadIdx.x, ty = threadIdx.y,
