commit 36522d991a13ebca18fb4cd64f554d3914113846
author Ben Vanik <benvanik@google.com> Thu Apr 29 07:39:40 2021 -0700
committer GitHub <noreply@github.com> Thu Apr 29 07:39:40 2021 -0700
tree 3db04c1a515a1d0cbddcf1e67b7e669914033bd5
parent e130e795a371c83e46022a9df093e4f2d4fd7c01

Rotating instead of shifting to deal with overflow. (#5674)

iree/task/topology.c

diff --git a/iree/task/topology.c b/iree/task/topology.c
index e901528..d96b627 100644
--- a/iree/task/topology.c
+++ b/iree/task/topology.c
@@ -248,9 +248,10 @@
     for (iree_host_size_t j = 0; j < topology->group_count; ++j) {
       if (i == j) continue;
       const iree_task_topology_group_t* other_group = &topology->groups[j];
-      uint64_t group_processor_bits = 1ull << other_group->processor_index;
+      uint64_t group_processor_bits =
+          iree_math_rotl_u64(1ull, other_group->processor_index);
       if (constructive_sharing_mask & group_processor_bits) {
-        group_mask |= 1ull << other_group->group_index;
+        group_mask |= iree_math_rotl_u64(1ull, other_group->group_index);
       }
     }