libflame revision_anchor
Functions
FLA_Ttmm_u_opt_var2.c File Reference

(r)

Functions

FLA_Error FLA_Ttmm_u_opt_var2 (FLA_Obj A)
FLA_Error FLA_Ttmm_u_ops_var2 (int mn_A, float *buff_A, int rs_A, int cs_A)
FLA_Error FLA_Ttmm_u_opd_var2 (int mn_A, double *buff_A, int rs_A, int cs_A)
FLA_Error FLA_Ttmm_u_opc_var2 (int mn_A, scomplex *buff_A, int rs_A, int cs_A)
FLA_Error FLA_Ttmm_u_opz_var2 (int mn_A, dcomplex *buff_A, int rs_A, int cs_A)

Function Documentation

FLA_Error FLA_Ttmm_u_opc_var2 ( int  mn_A,
scomplex buff_A,
int  rs_A,
int  cs_A 
)

References bli_cdots(), bli_cgemv(), bli_cscalv(), and FLA_ONE.

Referenced by FLA_Ttmm_u_opt_var2().

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

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

    int       mn_ahead  = mn_A - i - 1;
    int       mn_behind = i;

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

    // FLA_Scal_external( alpha11, a01 );
    bli_cscalv( BLIS_NO_CONJUGATE,
                mn_behind,
                alpha11,
                a01, rs_A );

    // FLA_Gemvc_external( FLA_NO_TRANSPOSE, FLA_CONJUGATE, FLA_ONE, A02, a12t, FLA_ONE, a01 );
    bli_cgemv( BLIS_NO_TRANSPOSE,
               BLIS_CONJUGATE,
               mn_behind,
               mn_ahead,
               buff_1,
               A02, rs_A, cs_A,
               a12t, cs_A,
               buff_1,
               a01, rs_A );

    // FLA_Absolute_square( alpha11 );
    bli_cabsqr( alpha11 );

    // FLA_Dotcs_external( FLA_CONJUGATE, FLA_ONE, a21, a21, FLA_ONE, alpha11 );
    bli_cdots( BLIS_CONJUGATE,
               mn_ahead,
               buff_1,
               a12t, cs_A,
               a12t, cs_A,
               buff_1,
               alpha11 );

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

  }

  return FLA_SUCCESS;
}
FLA_Error FLA_Ttmm_u_opd_var2 ( int  mn_A,
double *  buff_A,
int  rs_A,
int  cs_A 
)

References bli_ddots(), bli_dgemv(), bli_dscalv(), and FLA_ONE.

Referenced by FLA_Ttmm_u_opt_var2().

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

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

    int       mn_ahead  = mn_A - i - 1;
    int       mn_behind = i;

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

    // FLA_Scal_external( alpha11, a01 );
    bli_dscalv( BLIS_NO_CONJUGATE,
                mn_behind,
                alpha11,
                a01, rs_A );

    // FLA_Gemvc_external( FLA_NO_TRANSPOSE, FLA_CONJUGATE, FLA_ONE, A02, a12t, FLA_ONE, a01 );
    bli_dgemv( BLIS_NO_TRANSPOSE,
               BLIS_CONJUGATE,
               mn_behind,
               mn_ahead,
               buff_1,
               A02, rs_A, cs_A,
               a12t, cs_A,
               buff_1,
               a01, rs_A );

    // FLA_Absolute_square( alpha11 );
    bli_dabsqr( alpha11 );

    // FLA_Dotcs_external( FLA_CONJUGATE, FLA_ONE, a21, a21, FLA_ONE, alpha11 );
    bli_ddots( BLIS_CONJUGATE,
               mn_ahead,
               buff_1,
               a12t, cs_A,
               a12t, cs_A,
               buff_1,
               alpha11 );

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

  }

  return FLA_SUCCESS;
}
FLA_Error FLA_Ttmm_u_ops_var2 ( int  mn_A,
float *  buff_A,
int  rs_A,
int  cs_A 
)

References bli_sdots(), bli_sgemv(), bli_sscalv(), and FLA_ONE.

Referenced by FLA_Ttmm_u_opt_var2().

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

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

    int       mn_ahead  = mn_A - i - 1;
    int       mn_behind = i;

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

    // FLA_Scal_external( alpha11, a01 );
    bli_sscalv( BLIS_NO_CONJUGATE,
                mn_behind,
                alpha11,
                a01, rs_A );

    // FLA_Gemvc_external( FLA_NO_TRANSPOSE, FLA_CONJUGATE, FLA_ONE, A02, a12t, FLA_ONE, a01 );
    bli_sgemv( BLIS_NO_TRANSPOSE,
               BLIS_CONJUGATE,
               mn_behind,
               mn_ahead,
               buff_1,
               A02, rs_A, cs_A,
               a12t, cs_A,
               buff_1,
               a01, rs_A );

    // FLA_Absolute_square( alpha11 );
    bli_sabsqr( alpha11 );

    // FLA_Dotcs_external( FLA_CONJUGATE, FLA_ONE, a21, a21, FLA_ONE, alpha11 );
    bli_sdots( BLIS_CONJUGATE,
               mn_ahead,
               buff_1,
               a12t, cs_A,
               a12t, cs_A,
               buff_1,
               alpha11 );

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

  }

  return FLA_SUCCESS;
}
FLA_Error FLA_Ttmm_u_opt_var2 ( FLA_Obj  A)

References FLA_Obj_col_stride(), FLA_Obj_datatype(), FLA_Obj_length(), FLA_Obj_row_stride(), FLA_Ttmm_u_opc_var2(), FLA_Ttmm_u_opd_var2(), FLA_Ttmm_u_ops_var2(), and FLA_Ttmm_u_opz_var2().

Referenced by FLA_Ttmm_u().

{
  FLA_Datatype datatype;
  int          mn_A;
  int          rs_A, cs_A;

  datatype = FLA_Obj_datatype( A );

  mn_A     = FLA_Obj_length( A );
  rs_A     = FLA_Obj_row_stride( A );
  cs_A     = FLA_Obj_col_stride( A );
  

  switch ( datatype )
  {
    case FLA_FLOAT:
    {
      float* buff_A = FLA_FLOAT_PTR( A );

      FLA_Ttmm_u_ops_var2( mn_A,
                           buff_A, rs_A, cs_A );

      break;
    }

    case FLA_DOUBLE:
    {
      double* buff_A = FLA_DOUBLE_PTR( A );

      FLA_Ttmm_u_opd_var2( mn_A,
                           buff_A, rs_A, cs_A );

      break;
    }

    case FLA_COMPLEX:
    {
      scomplex* buff_A = FLA_COMPLEX_PTR( A );

      FLA_Ttmm_u_opc_var2( mn_A,
                           buff_A, rs_A, cs_A );

      break;
    }

    case FLA_DOUBLE_COMPLEX:
    {
      dcomplex* buff_A = FLA_DOUBLE_COMPLEX_PTR( A );

      FLA_Ttmm_u_opz_var2( mn_A,
                           buff_A, rs_A, cs_A );

      break;
    }
  }

  return FLA_SUCCESS;
}
FLA_Error FLA_Ttmm_u_opz_var2 ( int  mn_A,
dcomplex buff_A,
int  rs_A,
int  cs_A 
)

References bli_zdots(), bli_zgemv(), bli_zscalv(), and FLA_ONE.

Referenced by FLA_Ttmm_u_opt_var2().

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

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

    int       mn_ahead  = mn_A - i - 1;
    int       mn_behind = i;

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

    // FLA_Scal_external( alpha11, a01 );
    bli_zscalv( BLIS_NO_CONJUGATE,
                mn_behind,
                alpha11,
                a01, rs_A );

    // FLA_Gemvc_external( FLA_NO_TRANSPOSE, FLA_CONJUGATE, FLA_ONE, A02, a12t, FLA_ONE, a01 );
    bli_zgemv( BLIS_NO_TRANSPOSE,
               BLIS_CONJUGATE,
               mn_behind,
               mn_ahead,
               buff_1,
               A02, rs_A, cs_A,
               a12t, cs_A,
               buff_1,
               a01, rs_A );

    // FLA_Absolute_square( alpha11 );
    bli_zabsqr( alpha11 );

    // FLA_Dotcs_external( FLA_CONJUGATE, FLA_ONE, a21, a21, FLA_ONE, alpha11 );
    bli_zdots( BLIS_CONJUGATE,
               mn_ahead,
               buff_1,
               a12t, cs_A,
               a12t, cs_A,
               buff_1,
               alpha11 );

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

  }

  return FLA_SUCCESS;
}