gh0s7/mckernel
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

HFI1: refactor sdma_select_user_engine()

Browse Source
This commit is contained in:
Balazs Gerofi
2018-02-07 19:28:52 +09:00
parent 880dd6ddb2
commit ec5328de69
1 changed files with 30 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

44
kernel/sdma.c
View File

@@ -50,6 +50,16 @@
#include <hfi1/sdma.h> #include <hfi1/sdma.h>
#include <hfi1/common.h> #include <hfi1/common.h>
//#define DEBUG_PRINT_SDMA
#ifdef DEBUG_PRINT_SC
#define dkprintf(...) kprintf(__VA_ARGS__)
#define ekprintf(...) kprintf(__VA_ARGS__)
#else
#define dkprintf(...) do { if (0) kprintf(__VA_ARGS__); } while (0)
#define ekprintf(...) kprintf(__VA_ARGS__)
#endif
unsigned long hfi1_cap_mask = HFI1_CAP_MASK_DEFAULT; unsigned long hfi1_cap_mask = HFI1_CAP_MASK_DEFAULT;
/* must be a power of 2 >= 64 <= 32768 */ /* must be a power of 2 >= 64 <= 32768 */
@@ -101,6 +111,25 @@ done:
return rval; return rval;
} }
int sdma_select_user_engine_idx(void)
{
int idx = 0;
int idx_start = 0;
int idx_modulo = 16;
/* Hash on rank if MPI job */
if (cpu_local_var(current)->proc->nr_processes > 1) {
idx = idx_start +
(cpu_local_var(current)->proc->process_rank % idx_modulo);
}
/* Otherwise, CPU id */
else {
idx = ihk_mc_get_processor_id() % idx_modulo;
}
return idx;
}
/* /*
* sdma_select_user_engine() - select sdma engine based on user setup * sdma_select_user_engine() - select sdma engine based on user setup
* @dd: devdata * @dd: devdata
@@ -115,20 +144,7 @@ done:
struct sdma_engine *sdma_select_user_engine(struct hfi1_devdata *dd, struct sdma_engine *sdma_select_user_engine(struct hfi1_devdata *dd,
u32 selector, u8 vl) u32 selector, u8 vl)
{ {
int idx = 0; return &dd->per_sdma[sdma_select_user_engine_idx()];
int idx_start = 0;
int idx_modulo = 16;
/* Hash on rank for MPI jobs */
if (cpu_local_var(current)->proc->nr_processes > 1) {
idx = idx_start +
(cpu_local_var(current)->proc->process_rank % idx_modulo);
}
else {
idx = ihk_mc_get_processor_id() % idx_modulo;
}
return &dd->per_sdma[idx];
} }
/* /*