- GROUP_AFFINITY affinity;
- if (fun2(group, &affinity))
- fun3(GetCurrentThread(), &affinity, nullptr);
+ if (!fun4) {
+ GROUP_AFFINITY affinity;
+ if (fun2(node, &affinity))
+ fun3(GetCurrentThread(), &affinity, nullptr);
+ } else {
+ // If a numa node has more than one processor group, we assume they are
+ // sized equal and we spread threads evenly across the groups.
+ USHORT elements, returnedElements;
+ elements = GetMaximumProcessorGroupCount();
+ GROUP_AFFINITY *affinity = (GROUP_AFFINITY*)malloc(
+ elements * sizeof(GROUP_AFFINITY));
+ if (fun4(node, affinity, elements, &returnedElements))
+ fun3(GetCurrentThread(), &affinity[idx % returnedElements], nullptr);
+ free(affinity);
+ }