libflame
12600
|
void FLASH_Hemm_cntl_finalize | ( | void | ) |
References FLA_Blocksize_free(), and FLA_Cntl_obj_free().
Referenced by FLA_Cntl_finalize_flash().
void FLASH_Hemm_cntl_init | ( | void | ) |
References FLA_Blocksize_create(), and FLA_Cntl_hemm_obj_create().
Referenced by FLA_Cntl_init_flash().
{ // Set hemm blocksize for hierarchical storage. flash_hemm_bsize = FLA_Blocksize_create( 1, 1, 1, 1 ); // Create a control tree that assumes A and B are b x b blocks. flash_hemm_cntl_blas = FLA_Cntl_hemm_obj_create( FLA_HIER, FLA_SUBPROBLEM, NULL, NULL, NULL, NULL, NULL ); // Create a control tree that assumes A is a block and B is a panel. flash_hemm_cntl_bp = FLA_Cntl_hemm_obj_create( FLA_HIER, FLA_BLOCKED_VARIANT9, flash_hemm_bsize, flash_scal_cntl, flash_hemm_cntl_blas, NULL, NULL ); // Create a control tree that assumes A is large and B is a panel. flash_hemm_cntl_mp = FLA_Cntl_hemm_obj_create( FLA_HIER, FLA_BLOCKED_VARIANT1, flash_hemm_bsize, flash_scal_cntl, flash_hemm_cntl_blas, flash_gemm_cntl_op_bp, flash_gemm_cntl_mm_mp ); // Create a control tree that assumes A and B are both large. flash_hemm_cntl_mm = FLA_Cntl_hemm_obj_create( FLA_HIER, FLA_BLOCKED_VARIANT9, flash_hemm_bsize, flash_scal_cntl, flash_hemm_cntl_mp, NULL, NULL ); }
Referenced by FLA_Hemm_internal().