amd/blis
1
0
Fork 0
mirror of https://github.com/amd/blis.git synced 2026-05-11 17:50:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixed a bug for TRSM when BLIS_ENABLE_MULTITHREADING is not set but the multithreading environment variables are turned on

Browse Source
This commit is contained in:
Tyler Smith
2014-07-03 11:19:43 -05:00
parent b8134b720b
commit ab3bc9153b
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
frame/3/trsm/bli_trsm_threading.c
View File

@@ -109,16 +109,20 @@ void bli_trsm_thrinfo_free_paths( trsm_thrinfo_t** threads, dim_t num )
trsm_thrinfo_t** bli_create_trsm_thrinfo_paths( )
{
#ifdef BLIS_ENABLE_MULTITHREADING
dim_t jc_in = bli_read_nway_from_env( "BLIS_JC_NT" );
/*dim_t kc_in = bli_read_nway_from_env( "BLIS_KC_NT" );*/
dim_t ic_in = bli_read_nway_from_env( "BLIS_IC_NT" );
dim_t jr_in = bli_read_nway_from_env( "BLIS_JR_NT" );
dim_t ir_in = bli_read_nway_from_env( "BLIS_IR_NT" );
dim_t jr_way = jc_in * ic_in * jr_in * ir_in;
#else
dim_t jr_way = 1;
#endif
dim_t jc_way = 1;
dim_t kc_way = 1;
dim_t ic_way = 1;
dim_t jr_way = jc_in * ic_in * jr_in * ir_in;
dim_t ir_way = 1;
dim_t global_num_threads = jc_way * kc_way * ic_way * jr_way * ir_way;