Merge branch 'alloc-tmem' into tmem-copy-kernel

NVIDIA · Jan 30, 2025 · f9adf69 · f9adf69
2 parents 8bcbcae + 12ccf82
commit f9adf69
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/csrc/device_lower/pass/allocation.cpp b/csrc/device_lower/pass/allocation.cpp
@@ -871,9 +871,10 @@ std::vector<Expr*> insertTMemRegionAllocsAndDeallocs(
         kir::Asm::Options{/*volatile=*/true});
     prologue.push_back(tcgen05_relinquish_expr);
 
-    // Block sync that makes allocation visible to all threads
-    auto block_sync = IrBuilder::create<kir::BlockSync>();
-    prologue.push_back(block_sync);
+      // Block sync that makes allocation visible to all threads
+      auto block_sync = IrBuilder::create<kir::BlockSync>();
+      prologue.push_back(block_sync);
+    }
   }
 
   // Add deallocations to existing expressions