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

Merge pull request #127 from devinamatthews/fix_blis_nt_xx

Browse Source 
Setting any one of BLIS_NT_[IJ][CR] overrides BLIS_NUM_THEADS
This commit is contained in:
Tyler Michael Smith
2017-05-08 11:21:20 -05:00
committed by GitHub
parent cf39d3ef3b fdc66f12d4
commit cbf8710a1b
1 changed files with 14 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
frame/base/bli_cntx.c
View File

@@ -784,12 +784,20 @@ void bli_cntx_set_thrloop_from_env( opid_t l3_op, side_t side, cntx_t* cntx,
}
}
jc = bli_env_read_nway( "BLIS_JC_NT", jc );
//pc = bli_env_read_nway( "BLIS_KC_NT", 1 );
pc = 1;
ic = bli_env_read_nway( "BLIS_IC_NT", ic );
jr = bli_env_read_nway( "BLIS_JR_NT", jr );
ir = bli_env_read_nway( "BLIS_IR_NT", ir );
pc = 1;
dim_t jc_env = bli_env_read_nway( "BLIS_JC_NT", -1 );
dim_t ic_env = bli_env_read_nway( "BLIS_IC_NT", -1 );
dim_t jr_env = bli_env_read_nway( "BLIS_JR_NT", -1 );
dim_t ir_env = bli_env_read_nway( "BLIS_IR_NT", -1 );
if (jc_env != -1 || ic_env != -1 || jr_env != -1 || ir_env != -1)
{
jc = (jc_env == -1 ? 1 : jc_env);
ic = (ic_env == -1 ? 1 : ic_env);
jr = (jr_env == -1 ? 1 : jr_env);
ir = (ir_env == -1 ? 1 : ir_env);
}
#else