xe: ir: walk order: merge consecutive blocks

oneapi-src · Oct 18, 2024 · 7e1663f · 7e1663f
1 parent 1794c34
commit 7e1663f
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/gpu/intel/jit/ir/walk_order.hpp b/src/gpu/intel/jit/ir/walk_order.hpp
@@ -62,6 +62,13 @@ class walk_order_t {
     }
 
     void add(const prb_dim_t &dim, int block_size, int grid_id) {
+        if (!blocks_.empty()) {
+            auto &last = blocks_.back();
+            if (last.dim == dim && last.grid_id == grid_id) {
+                last.size *= block_size;
+                return;
+            }
+        }
         blocks_.emplace_back(dim, block_size, grid_id);
     }