50cents readibility improvement to allocgroups on GPU schedules.

halide · Oct 6, 2023 · 92339f2 · 92339f2
1 parent 24a64f8
commit 92339f2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/FuseGPUThreadLoops.cpp b/src/FuseGPUThreadLoops.cpp
@@ -802,14 +802,14 @@ class ExtractSharedAndHeapAllocations : public IRMutator {
             Type widest_type;
             for (const auto &alloc : cluster) {
                 if (name.empty()) {
-                    name = alloc.name;
+                    name = "allocgroup";
                     widest_type = alloc.widest_type;
                 } else {
-                    name += "__" + alloc.name;
                     if (alloc.widest_type.bytes() > widest_type.bytes()) {
                         widest_type = alloc.widest_type;
                     }
                 }
+                name += "__" + alloc.name;
                 int ratio = alloc.widest_type.bytes() / alloc_type.bytes();
                 internal_assert(ratio != 0)
                     << "alloc_type should have been at most as wide as the widest type in group\n";