Fix assertion in reorder_communication_preserving_peak_memory (#152565)

wconstab · pytorchmergebot · commit 53bf17462660 · 2025-05-01T06:40:04.000Z
>=0 is practically correct becuase we do model the runtime of some ops as 0. Pull Request resolved: #152565 Approved by: https://github.com/eellison
diff --git a/torch/_inductor/comms.py b/torch/_inductor/comms.py
@@ -165,7 +165,7 @@ class ReorderInfo:
 
     @property
     def improvement(self):
-        assert self.initial_exposed >= self.final_exposed > 0
+        assert self.initial_exposed >= self.final_exposed >= 0
         return self.initial_exposed - self.final_exposed
 
 
