libflame  revision_anchor
Functions
FLA_Chol.h File Reference

(r)

Go to the source code of this file.

Functions

FLA_Error FLA_Chol_internal (FLA_Uplo uplo, FLA_Obj A, fla_chol_t *cntl)
FLA_Error FLA_Chol_l (FLA_Obj A, fla_chol_t *cntl)
FLA_Error FLA_Chol_u (FLA_Obj A, fla_chol_t *cntl)
FLA_Error FLA_Chol_solve (FLA_Uplo uplo, FLA_Obj A, FLA_Obj B, FLA_Obj X)
FLA_Error FLASH_Chol_solve (FLA_Uplo uplo, FLA_Obj A, FLA_Obj B, FLA_Obj X)

Function Documentation

FLA_Error FLA_Chol_internal ( FLA_Uplo  uplo,
FLA_Obj  A,
fla_chol_t cntl 
)

References FLA_Check_error_level(), fla_chol_cntl_leaf, FLA_Chol_internal(), FLA_Chol_internal_check(), FLA_Chol_l(), FLA_Chol_u(), FLA_Obj_elemtype(), and FLASH_Queue_get_enabled().

Referenced by FLA_Chol(), FLA_Chol_internal(), FLA_Chol_l_blk_var1(), FLA_Chol_l_blk_var2(), FLA_Chol_l_blk_var3(), FLA_Chol_l_task(), FLA_Chol_task(), FLA_Chol_u_blk_var1(), FLA_Chol_u_blk_var2(), FLA_Chol_u_blk_var3(), FLA_Chol_u_task(), FLA_SPDinv_internal(), and FLASH_Chol().

{
    FLA_Error r_val = FLA_SUCCESS;

    if ( FLA_Check_error_level() == FLA_FULL_ERROR_CHECKING )
        FLA_Chol_internal_check( uplo, A, cntl );

    if      ( FLA_Cntl_matrix_type( cntl ) == FLA_HIER &&
              FLA_Obj_elemtype( A ) == FLA_MATRIX &&
              FLA_Cntl_variant( cntl ) == FLA_SUBPROBLEM )
    {
        // Recurse
        r_val = FLA_Chol_internal( uplo,
                                   *FLASH_OBJ_PTR_AT( A ),
                                   flash_chol_cntl );
    }
    else if ( FLA_Cntl_matrix_type( cntl ) == FLA_HIER &&
              FLA_Obj_elemtype( A ) == FLA_SCALAR &&
              FLASH_Queue_get_enabled( ) )
    {
        // Enqueue
        ENQUEUE_FLASH_Chol( uplo, A, cntl );
    }
    else
    {
        if ( FLA_Cntl_matrix_type( cntl ) == FLA_HIER &&
             FLA_Obj_elemtype( A ) == FLA_SCALAR &&
             !FLASH_Queue_get_enabled( ) )
        {
            // Execute leaf
            cntl = fla_chol_cntl_leaf;
        }

        // Parameter combinations
        if      ( uplo == FLA_LOWER_TRIANGULAR )
        {
            r_val = FLA_Chol_l( A, cntl );
        }
        else if ( uplo == FLA_UPPER_TRIANGULAR )
        {
            r_val = FLA_Chol_u( A, cntl );
        }
    }

    return r_val;
}
FLA_Error FLA_Chol_l ( FLA_Obj  A,
fla_chol_t cntl 
)

References FLA_Chol_l_blk_ext(), FLA_Chol_l_blk_var1(), FLA_Chol_l_blk_var2(), FLA_Chol_l_blk_var3(), FLA_Chol_l_opt_var1(), FLA_Chol_l_opt_var2(), FLA_Chol_l_opt_var3(), FLA_Chol_l_unb_ext(), FLA_Chol_l_unb_var1(), FLA_Chol_l_unb_var2(), and FLA_Chol_l_unb_var3().

Referenced by FLA_Chol_internal().

{
    FLA_Error r_val = FLA_SUCCESS;
    
    if      ( FLA_Cntl_variant( cntl ) == FLA_BLOCKED_EXTERN )
    {
        r_val = FLA_Chol_l_blk_ext( A );
    }
    else if ( FLA_Cntl_variant( cntl ) == FLA_UNBLOCKED_EXTERN )
    {
        r_val = FLA_Chol_l_unb_ext( A );
    }
#ifdef FLA_ENABLE_NON_CRITICAL_CODE
    else if ( FLA_Cntl_variant( cntl ) == FLA_UNB_OPT_VARIANT1 )
    {
        r_val = FLA_Chol_l_opt_var1( A );
    }
#endif
    else if ( FLA_Cntl_variant( cntl ) == FLA_UNB_OPT_VARIANT2 )
    {
        r_val = FLA_Chol_l_opt_var2( A );
    }
    else if ( FLA_Cntl_variant( cntl ) == FLA_UNB_OPT_VARIANT3 )
    {
        r_val = FLA_Chol_l_opt_var3( A );
    }
#ifdef FLA_ENABLE_NON_CRITICAL_CODE
    else if ( FLA_Cntl_variant( cntl ) == FLA_BLOCKED_VARIANT1 )
    {
        r_val = FLA_Chol_l_blk_var1( A, cntl );
    }
    else if ( FLA_Cntl_variant( cntl ) == FLA_BLOCKED_VARIANT2 )
    {
        r_val = FLA_Chol_l_blk_var2( A, cntl );
    }
#endif
    else if ( FLA_Cntl_variant( cntl ) == FLA_BLOCKED_VARIANT3 )
    {
        r_val = FLA_Chol_l_blk_var3( A, cntl );
    }
#ifdef FLA_ENABLE_NON_CRITICAL_CODE
    else if ( FLA_Cntl_variant( cntl ) == FLA_UNBLOCKED_VARIANT1 )
    {
        r_val = FLA_Chol_l_unb_var1( A );
    }
    else if ( FLA_Cntl_variant( cntl ) == FLA_UNBLOCKED_VARIANT2 )
    {
        r_val = FLA_Chol_l_unb_var2( A );
    }
    else if ( FLA_Cntl_variant( cntl ) == FLA_UNBLOCKED_VARIANT3 )
    {
        r_val = FLA_Chol_l_unb_var3( A );
    }
#endif
    else
    {
        FLA_Check_error_code( FLA_NOT_YET_IMPLEMENTED );
    }

    return r_val;
}
FLA_Error FLA_Chol_solve ( FLA_Uplo  uplo,
FLA_Obj  A,
FLA_Obj  B,
FLA_Obj  X 
)

References FLA_Check_error_level(), FLA_Chol_solve_check(), FLA_Copy_external(), FLA_ONE, and FLA_Trsm_external().

{
  // Check parameters.
  if ( FLA_Check_error_level() >= FLA_MIN_ERROR_CHECKING )
    FLA_Chol_solve_check( uplo, A, B, X );

  FLA_Copy_external( B, X );

  if ( uplo == FLA_LOWER_TRIANGULAR )
  {
      FLA_Trsm_external( FLA_LEFT, FLA_LOWER_TRIANGULAR, FLA_NO_TRANSPOSE,
                         FLA_NONUNIT_DIAG, FLA_ONE, A, X );
      FLA_Trsm_external( FLA_LEFT, FLA_LOWER_TRIANGULAR, FLA_CONJ_TRANSPOSE,
                         FLA_NONUNIT_DIAG, FLA_ONE, A, X );
  }
  else // if ( uplo == FLA_UPPER_TRIANGULAR )
  {
      FLA_Trsm_external( FLA_LEFT, FLA_UPPER_TRIANGULAR, FLA_CONJ_TRANSPOSE,
                         FLA_NONUNIT_DIAG, FLA_ONE, A, X );
      FLA_Trsm_external( FLA_LEFT, FLA_UPPER_TRIANGULAR, FLA_NO_TRANSPOSE,
                         FLA_NONUNIT_DIAG, FLA_ONE, A, X );
  }

  return FLA_SUCCESS;
}
FLA_Error FLA_Chol_u ( FLA_Obj  A,
fla_chol_t cntl 
)

References FLA_Chol_u_blk_ext(), FLA_Chol_u_blk_var1(), FLA_Chol_u_blk_var2(), FLA_Chol_u_blk_var3(), FLA_Chol_u_opt_var1(), FLA_Chol_u_opt_var2(), FLA_Chol_u_opt_var3(), FLA_Chol_u_unb_ext(), FLA_Chol_u_unb_var1(), FLA_Chol_u_unb_var2(), and FLA_Chol_u_unb_var3().

Referenced by FLA_Chol_internal().

{
    FLA_Error r_val = FLA_SUCCESS;
    
    if      ( FLA_Cntl_variant( cntl ) == FLA_BLOCKED_EXTERN )
    {
        r_val = FLA_Chol_u_blk_ext( A );
    }
    else if ( FLA_Cntl_variant( cntl ) == FLA_UNBLOCKED_EXTERN )
    {
        r_val = FLA_Chol_u_unb_ext( A );
    }
#ifdef FLA_ENABLE_NON_CRITICAL_CODE
    else if ( FLA_Cntl_variant( cntl ) == FLA_UNB_OPT_VARIANT1 )
    {
        r_val = FLA_Chol_u_opt_var1( A );
    }
#endif
    else if ( FLA_Cntl_variant( cntl ) == FLA_UNB_OPT_VARIANT2 )
    {
        r_val = FLA_Chol_u_opt_var2( A );
    }
    else if ( FLA_Cntl_variant( cntl ) == FLA_UNB_OPT_VARIANT3 )
    {
        r_val = FLA_Chol_u_opt_var3( A );
    }
#ifdef FLA_ENABLE_NON_CRITICAL_CODE
    else if ( FLA_Cntl_variant( cntl ) == FLA_BLOCKED_VARIANT1 )
    {
        r_val = FLA_Chol_u_blk_var1( A, cntl );
    }
    else if ( FLA_Cntl_variant( cntl ) == FLA_BLOCKED_VARIANT2 )
    {
        r_val = FLA_Chol_u_blk_var2( A, cntl );
    }
#endif
    else if ( FLA_Cntl_variant( cntl ) == FLA_BLOCKED_VARIANT3 )
    {
        r_val = FLA_Chol_u_blk_var3( A, cntl );
    }
#ifdef FLA_ENABLE_NON_CRITICAL_CODE
    else if ( FLA_Cntl_variant( cntl ) == FLA_UNBLOCKED_VARIANT1 )
    {
        r_val = FLA_Chol_u_unb_var1( A );
    }
    else if ( FLA_Cntl_variant( cntl ) == FLA_UNBLOCKED_VARIANT2 )
    {
        r_val = FLA_Chol_u_unb_var2( A );
    }
    else if ( FLA_Cntl_variant( cntl ) == FLA_UNBLOCKED_VARIANT3 )
    {
        r_val = FLA_Chol_u_unb_var3( A );
    }
#endif
    else
    {
        FLA_Check_error_code( FLA_NOT_YET_IMPLEMENTED );
    }

    return r_val;
}

References FLA_Check_error_level(), FLA_Chol_solve_check(), FLA_ONE, FLASH_Copy(), and FLASH_Trsm().

{
  // Check parameters.
  if ( FLA_Check_error_level() >= FLA_MIN_ERROR_CHECKING )
    FLA_Chol_solve_check( uplo, A, B, X );

  FLASH_Copy( B, X );

  if ( uplo == FLA_LOWER_TRIANGULAR )
  {
      FLASH_Trsm( FLA_LEFT, FLA_LOWER_TRIANGULAR, FLA_NO_TRANSPOSE,
                  FLA_NONUNIT_DIAG, FLA_ONE, A, X );
      FLASH_Trsm( FLA_LEFT, FLA_LOWER_TRIANGULAR, FLA_CONJ_TRANSPOSE,
                  FLA_NONUNIT_DIAG, FLA_ONE, A, X );
  }
  else // if ( uplo == FLA_UPPER_TRIANGULAR )
  {
      FLASH_Trsm( FLA_LEFT, FLA_UPPER_TRIANGULAR, FLA_CONJ_TRANSPOSE,
                  FLA_NONUNIT_DIAG, FLA_ONE, A, X );
      FLASH_Trsm( FLA_LEFT, FLA_UPPER_TRIANGULAR, FLA_NO_TRANSPOSE,
                  FLA_NONUNIT_DIAG, FLA_ONE, A, X );
  }

  return FLA_SUCCESS;
}