libflame revision_anchor
Functions
FLA_Accum_T_UT_fr.h File Reference

(r)

Go to the source code of this file.

Functions

FLA_Error FLA_Accum_T_UT_fr_unb_var1 (FLA_Obj A, FLA_Obj t, FLA_Obj T)
FLA_Error FLA_Accum_T_UT_fr_blk_var2 (FLA_Obj A, FLA_Obj t, FLA_Obj T)
FLA_Error FLA_Accum_T_UT_fr_opt_var1 (FLA_Obj A, FLA_Obj t, FLA_Obj T)
FLA_Error FLA_Accum_T_UT_fr_ops_var1 (int m_A, int n_A, float *A, int rs_A, int cs_A, float *t, int inc_t, float *T, int rs_T, int cs_T)
FLA_Error FLA_Accum_T_UT_fr_opd_var1 (int m_A, int n_A, double *A, int rs_A, int cs_A, double *t, int inc_t, double *T, int rs_T, int cs_T)
FLA_Error FLA_Accum_T_UT_fr_opc_var1 (int m_A, int n_A, scomplex *A, int rs_A, int cs_A, scomplex *t, int inc_t, scomplex *T, int rs_T, int cs_T)
FLA_Error FLA_Accum_T_UT_fr_opz_var1 (int m_A, int n_A, dcomplex *A, int rs_A, int cs_A, dcomplex *t, int inc_t, dcomplex *T, int rs_T, int cs_T)

Function Documentation

FLA_Error FLA_Accum_T_UT_fr_blk_var2 ( FLA_Obj  A,
FLA_Obj  t,
FLA_Obj  T 
)

References FLA_Accum_T_UT_fr_opt_var1(), FLA_Cont_with_1x3_to_1x2(), FLA_Cont_with_3x1_to_2x1(), FLA_Cont_with_3x3_to_2x2(), FLA_Merge_1x2(), FLA_Obj_length(), FLA_Obj_width(), FLA_Part_1x2(), FLA_Part_2x1(), FLA_Part_2x2(), FLA_Repart_1x2_to_1x3(), FLA_Repart_2x1_to_3x1(), and FLA_Repart_2x2_to_3x3().

Referenced by FLA_Accum_T_UT_internal().

{
  FLA_Obj ATL,   ATR,      A00, A01, A02, 
          ABL,   ABR,      A10, A11, A12,
                           A20, A21, A22;

  FLA_Obj tT,              t0,
          tB,              t1,
                           t2;

  FLA_Obj TL,    TR,       T0,  T1,  T2;

  FLA_Obj AR1;

  dim_t b_alg, b;

  b_alg = FLA_Obj_length( T );

  FLA_Part_2x2( A,    &ATL, &ATR,
                      &ABL, &ABR,     0, 0, FLA_TL );

  FLA_Part_2x1( t,    &tT, 
                      &tB,            0, FLA_TOP );

  FLA_Part_1x2( T,    &TL,  &TR,      0, FLA_LEFT );

  while ( FLA_Obj_width( TL ) < FLA_Obj_width( T ) ){

    b = min( FLA_Obj_width( TR ), b_alg );

    FLA_Repart_2x2_to_3x3( ATL, /**/ ATR,       &A00, /**/ &A01, &A02,
                        /* ************* */   /* ******************** */
                                                &A10, /**/ &A11, &A12,
                           ABL, /**/ ABR,       &A20, /**/ &A21, &A22,
                           b, b, FLA_BR );

    FLA_Repart_2x1_to_3x1( tT,                &t0, 
                        /* ** */            /* ** */
                                              &t1, 
                           tB,                &t2,        b, FLA_BOTTOM );

    FLA_Repart_1x2_to_1x3( TL,  /**/ TR,        &T0, /**/ &T1, &T2,
                           b, FLA_RIGHT );

    /*------------------------------------------------------------*/

    FLA_Merge_1x2( A11, A12,   &AR1 );

    //FLA_Accum_T_UT_fr_unb_var1( AR1, t1, T1 );
    FLA_Accum_T_UT_fr_opt_var1( AR1, t1, T1 );

    /*------------------------------------------------------------*/

    FLA_Cont_with_3x3_to_2x2( &ATL, /**/ &ATR,       A00, A01, /**/ A02,
                                                     A10, A11, /**/ A12,
                            /* ************** */  /* ****************** */
                              &ABL, /**/ &ABR,       A20, A21, /**/ A22,
                              FLA_TL );

    FLA_Cont_with_3x1_to_2x1( &tT,                t0, 
                                                  t1, 
                            /* ** */           /* ** */
                              &tB,                t2,     FLA_TOP );

    FLA_Cont_with_1x3_to_1x2( &TL,  /**/ &TR,        T0, T1, /**/ T2,
                              FLA_LEFT );

  }

  return FLA_SUCCESS;
}
FLA_Error FLA_Accum_T_UT_fr_opc_var1 ( int  m_A,
int  n_A,
scomplex A,
int  rs_A,
int  cs_A,
scomplex t,
int  inc_t,
scomplex T,
int  rs_T,
int  cs_T 
)

References bli_ccopyv(), bli_cgemv(), and FLA_ONE.

Referenced by FLA_Accum_T_UT_fr_opt_var1().

{
  scomplex* buff_1   = FLA_COMPLEX_PTR( FLA_ONE );
  int       i;

  for ( i = 0; i < m_A; ++i )
  {
    scomplex* a01      = buff_A + (i  )*cs_A + (0  )*rs_A;
    scomplex* A02      = buff_A + (i+1)*cs_A + (0  )*rs_A;
    scomplex* a12t     = buff_A + (i+1)*cs_A + (i  )*rs_A;

    scomplex* tau1     = buff_t + (i  )*inc_t;

    scomplex* tau11    = buff_T + (i  )*cs_T + (i  )*rs_T;
    scomplex* t01      = buff_T + (i  )*cs_T + (0  )*rs_T;

    int       n_ahead  = n_A - i - 1;
    int       m_behind = i;

    /*------------------------------------------------------------*/

    // FLA_Copy_external( tau1, tau11 );
    *tau11 = *tau1;

    // FLA_Copyt_external( FLA_CONJ_NO_TRANSPOSE, a01, t01 );
    bli_ccopyv( BLIS_CONJUGATE,
                m_behind,
                a01, rs_A,
                t01, rs_T );

    // FLA_Gemvc_external( FLA_CONJ_NO_TRANSPOSE, FLA_NO_CONJUGATE,
    //                     FLA_ONE, A02, a12t, FLA_ONE, t01 );
    bli_cgemv( BLIS_CONJ_NO_TRANSPOSE,
               BLIS_NO_CONJUGATE,
               m_behind,
               n_ahead,
               buff_1,
               A02, rs_A, cs_A,
               a12t, cs_A,
               buff_1,
               t01, rs_T );

    /*------------------------------------------------------------*/

  }

  return FLA_SUCCESS;
}
FLA_Error FLA_Accum_T_UT_fr_opd_var1 ( int  m_A,
int  n_A,
double *  A,
int  rs_A,
int  cs_A,
double *  t,
int  inc_t,
double *  T,
int  rs_T,
int  cs_T 
)

References bli_dcopyv(), bli_dgemv(), and FLA_ONE.

Referenced by FLA_Accum_T_UT_fr_opt_var1().

{
  double* buff_1  = FLA_DOUBLE_PTR( FLA_ONE );
  int     i;

  for ( i = 0; i < m_A; ++i )
  {
    double* a01     = buff_A + (i  )*cs_A + (0  )*rs_A;
    double* A02     = buff_A + (i+1)*cs_A + (0  )*rs_A;
    double* a12t    = buff_A + (i+1)*cs_A + (i  )*rs_A;

    double* tau1    = buff_t + (i  )*inc_t;

    double* tau11   = buff_T + (i  )*cs_T + (i  )*rs_T;
    double* t01     = buff_T + (i  )*cs_T + (0  )*rs_T;

    int    n_ahead  = n_A - i - 1;
    int    m_behind = i;

    /*------------------------------------------------------------*/

    // FLA_Copy_external( tau1, tau11 );
    *tau11 = *tau1;

    // FLA_Copyt_external( FLA_CONJ_NO_TRANSPOSE, a01, t01 );
    bli_dcopyv( BLIS_CONJUGATE,
                m_behind,
                a01, rs_A,
                t01, rs_T );

    // FLA_Gemvc_external( FLA_CONJ_NO_TRANSPOSE, FLA_NO_CONJUGATE,
    //                     FLA_ONE, A02, a12t, FLA_ONE, t01 );
    bli_dgemv( BLIS_CONJ_NO_TRANSPOSE,
               BLIS_NO_CONJUGATE,
               m_behind,
               n_ahead,
               buff_1,
               A02, rs_A, cs_A,
               a12t, cs_A,
               buff_1,
               t01, rs_T );

    /*------------------------------------------------------------*/

  }

  return FLA_SUCCESS;
}
FLA_Error FLA_Accum_T_UT_fr_ops_var1 ( int  m_A,
int  n_A,
float *  A,
int  rs_A,
int  cs_A,
float *  t,
int  inc_t,
float *  T,
int  rs_T,
int  cs_T 
)

References bli_scopyv(), bli_sgemv(), and FLA_ONE.

Referenced by FLA_Accum_T_UT_fr_opt_var1().

{
  float* buff_1   = FLA_FLOAT_PTR( FLA_ONE );
  int    i;

  for ( i = 0; i < m_A; ++i )
  {
    float* a01      = buff_A + (i  )*cs_A + (0  )*rs_A;
    float* A02      = buff_A + (i+1)*cs_A + (0  )*rs_A;
    float* a12t     = buff_A + (i+1)*cs_A + (i  )*rs_A;

    float* tau1     = buff_t + (i  )*inc_t;

    float* tau11    = buff_T + (i  )*cs_T + (i  )*rs_T;
    float* t01      = buff_T + (i  )*cs_T + (0  )*rs_T;

    int    n_ahead  = n_A - i - 1;
    int    m_behind = i;

    /*------------------------------------------------------------*/

    // FLA_Copy_external( tau1, tau11 );
    *tau11 = *tau1;

    // FLA_Copyt_external( FLA_CONJ_NO_TRANSPOSE, a01, t01 );
    bli_scopyv( BLIS_CONJUGATE,
                m_behind,
                a01, rs_A,
                t01, rs_T );
  
    // FLA_Gemvc_external( FLA_CONJ_NO_TRANSPOSE, FLA_NO_CONJUGATE,
    //                     FLA_ONE, A02, a12t, FLA_ONE, t01 );
    bli_sgemv( BLIS_CONJ_NO_TRANSPOSE,
               BLIS_NO_CONJUGATE,
               m_behind,
               n_ahead,
               buff_1,
               A02, rs_A, cs_A,
               a12t, cs_A,
               buff_1,
               t01, rs_T );

    /*------------------------------------------------------------*/

  }

  return FLA_SUCCESS;
}
FLA_Error FLA_Accum_T_UT_fr_opt_var1 ( FLA_Obj  A,
FLA_Obj  t,
FLA_Obj  T 
)

References FLA_Accum_T_UT_fr_opc_var1(), FLA_Accum_T_UT_fr_opd_var1(), FLA_Accum_T_UT_fr_ops_var1(), FLA_Accum_T_UT_fr_opz_var1(), FLA_Obj_col_stride(), FLA_Obj_datatype(), FLA_Obj_length(), FLA_Obj_row_stride(), FLA_Obj_vector_inc(), and FLA_Obj_width().

Referenced by FLA_Accum_T_UT_fr_blk_var2().

{
  FLA_Datatype datatype;
  int          m_A, n_A;
  int          rs_A, cs_A;
  int          inc_t;
  int          rs_T, cs_T;

  datatype = FLA_Obj_datatype( A );

  m_A      = FLA_Obj_length( A );
  n_A      = FLA_Obj_width( A );
  rs_A     = FLA_Obj_row_stride( A );
  cs_A     = FLA_Obj_col_stride( A );

  inc_t    = FLA_Obj_vector_inc( t );

  rs_T     = FLA_Obj_row_stride( T );
  cs_T     = FLA_Obj_col_stride( T );

  switch ( datatype )
  {
    case FLA_FLOAT:
    {
      float* buff_A = FLA_FLOAT_PTR( A );
      float* buff_t = FLA_FLOAT_PTR( t );
      float* buff_T = FLA_FLOAT_PTR( T );

      FLA_Accum_T_UT_fr_ops_var1( m_A,
                                  n_A,
                                  buff_A, rs_A, cs_A,
                                  buff_t, inc_t,
                                  buff_T, rs_T, cs_T );

      break;
    }

    case FLA_DOUBLE:
    {
      double* buff_A = FLA_DOUBLE_PTR( A );
      double* buff_t = FLA_DOUBLE_PTR( t );
      double* buff_T = FLA_DOUBLE_PTR( T );

      FLA_Accum_T_UT_fr_opd_var1( m_A,
                                  n_A,
                                  buff_A, rs_A, cs_A,
                                  buff_t, inc_t,
                                  buff_T, rs_T, cs_T );

      break;
    }

    case FLA_COMPLEX:
    {
      scomplex* buff_A = FLA_COMPLEX_PTR( A );
      scomplex* buff_t = FLA_COMPLEX_PTR( t );
      scomplex* buff_T = FLA_COMPLEX_PTR( T );

      FLA_Accum_T_UT_fr_opc_var1( m_A,
                                  n_A,
                                  buff_A, rs_A, cs_A,
                                  buff_t, inc_t,
                                  buff_T, rs_T, cs_T );

      break;
    }

    case FLA_DOUBLE_COMPLEX:
    {
      dcomplex* buff_A = FLA_DOUBLE_COMPLEX_PTR( A );
      dcomplex* buff_t = FLA_DOUBLE_COMPLEX_PTR( t );
      dcomplex* buff_T = FLA_DOUBLE_COMPLEX_PTR( T );

      FLA_Accum_T_UT_fr_opz_var1( m_A,
                                  n_A,
                                  buff_A, rs_A, cs_A,
                                  buff_t, inc_t,
                                  buff_T, rs_T, cs_T );

      break;
    }
  }

  return FLA_SUCCESS;
}
FLA_Error FLA_Accum_T_UT_fr_opz_var1 ( int  m_A,
int  n_A,
dcomplex A,
int  rs_A,
int  cs_A,
dcomplex t,
int  inc_t,
dcomplex T,
int  rs_T,
int  cs_T 
)

References bli_zcopyv(), bli_zgemv(), and FLA_ONE.

Referenced by FLA_Accum_T_UT_fr_opt_var1().

{
  dcomplex* buff_1   = FLA_DOUBLE_COMPLEX_PTR( FLA_ONE );
  int       i;

  for ( i = 0; i < m_A; ++i )
  {
    dcomplex* a01      = buff_A + (i  )*cs_A + (0  )*rs_A;
    dcomplex* A02      = buff_A + (i+1)*cs_A + (0  )*rs_A;
    dcomplex* a12t     = buff_A + (i+1)*cs_A + (i  )*rs_A;

    dcomplex* tau1     = buff_t + (i  )*inc_t;

    dcomplex* tau11    = buff_T + (i  )*cs_T + (i  )*rs_T;
    dcomplex* t01      = buff_T + (i  )*cs_T + (0  )*rs_T;

    int       n_ahead  = n_A - i - 1;
    int       m_behind = i;

    /*------------------------------------------------------------*/

    // FLA_Copy_external( tau1, tau11 );
    *tau11 = *tau1;

    // FLA_Copyt_external( FLA_CONJ_NO_TRANSPOSE, a01, t01 );
    bli_zcopyv( BLIS_CONJUGATE,
                m_behind,
                a01, rs_A,
                t01, rs_T );

    // FLA_Gemvc_external( FLA_CONJ_NO_TRANSPOSE, FLA_NO_CONJUGATE,
    //                     FLA_ONE, A02, a12t, FLA_ONE, t01 );
    bli_zgemv( BLIS_CONJ_NO_TRANSPOSE,
               BLIS_NO_CONJUGATE,
               m_behind,
               n_ahead,
               buff_1,
               A02, rs_A, cs_A,
               a12t, cs_A,
               buff_1,
               t01, rs_T );

    /*------------------------------------------------------------*/

  }


  return FLA_SUCCESS;
}
FLA_Error FLA_Accum_T_UT_fr_unb_var1 ( FLA_Obj  A,
FLA_Obj  t,
FLA_Obj  T 
)

References FLA_Cont_with_3x1_to_2x1(), FLA_Cont_with_3x3_to_2x2(), FLA_Copy_external(), FLA_Copyt_external(), FLA_Gemvc_external(), FLA_Obj_min_dim(), FLA_ONE, FLA_Part_2x1(), FLA_Part_2x2(), FLA_Repart_2x1_to_3x1(), and FLA_Repart_2x2_to_3x3().

{
  FLA_Obj ATL,   ATR,      A00,  a01,     A02, 
          ABL,   ABR,      a10t, alpha11, a12t,
                           A20,  a21,     A22;

  FLA_Obj tauT,            tau0,
          tauB,            tau1,
                           tau2;

  FLA_Obj TTL,   TTR,      T00,  t01,   T02, 
          TBL,   TBR,      t10t, tau11, t12t,
                           T20,  t21,   T22;


  FLA_Part_2x2( A,    &ATL, &ATR,
                      &ABL, &ABR,     0, 0, FLA_TL );

  FLA_Part_2x1( tau,  &tauT, 
                      &tauB,          0, FLA_TOP );

  FLA_Part_2x2( T,    &TTL, &TTR,
                      &TBL, &TBR,     0, 0, FLA_TL );

  while ( FLA_Obj_min_dim( ABR ) > 0 ){

    FLA_Repart_2x2_to_3x3( ATL, /**/ ATR,       &A00,  /**/ &a01,     &A02,
                        /* ************* */   /* ************************** */
                                                &a10t, /**/ &alpha11, &a12t,
                           ABL, /**/ ABR,       &A20,  /**/ &a21,     &A22,
                           1, 1, FLA_BR );

    FLA_Repart_2x1_to_3x1( tauT,                &tau0, 
                        /* ** */              /* ** */
                                                &tau1, 
                           tauB,                &tau2,        1, FLA_BOTTOM );

    FLA_Repart_2x2_to_3x3( TTL, /**/ TTR,       &T00,  /**/ &t01,   &T02,
                        /* ************* */   /* ************************ */
                                                &t10t, /**/ &tau11, &t12t,
                           TBL, /**/ TBR,       &T20,  /**/ &t21,   &T22,
                           1, 1, FLA_BR );

    /*------------------------------------------------------------*/

    // tau11 = tau1;
    FLA_Copy_external( tau1, tau11 );

    // t01 = conj(a01) + conj(A02) * u12t^T;
    FLA_Copyt_external( FLA_CONJ_NO_TRANSPOSE, a01, t01 );
    FLA_Gemvc_external( FLA_CONJ_NO_TRANSPOSE, FLA_NO_CONJUGATE, FLA_ONE, A02, a12t, FLA_ONE, t01 );

    /*------------------------------------------------------------*/

    FLA_Cont_with_3x3_to_2x2( &ATL, /**/ &ATR,       A00,  a01,     /**/ A02,
                                                     a10t, alpha11, /**/ a12t,
                            /* ************** */  /* ************************ */
                              &ABL, /**/ &ABR,       A20,  a21,     /**/ A22,
                              FLA_TL );

    FLA_Cont_with_3x1_to_2x1( &tauT,                tau0, 
                                                    tau1, 
                            /* ** */              /* ** */
                              &tauB,                tau2,     FLA_TOP );

    FLA_Cont_with_3x3_to_2x2( &TTL, /**/ &TTR,       T00,  t01,   /**/ T02,
                                                     t10t, tau11, /**/ t12t,
                            /* ************** */  /* ********************** */
                              &TBL, /**/ &TBR,       T20,  t21,   /**/ T22,
                              FLA_TL );

  }

  return FLA_SUCCESS;
}