Tensor liveness supported

Accelergy-Project · Oct 24, 2024 · 105b721 · 105b721
1 parent 66616e9
commit 105b721
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pytimeloop/fastfusion/pareto.py b/pytimeloop/fastfusion/pareto.py
@@ -126,8 +126,8 @@ def free_to_loop_index(self, n: int) -> "Pareto":
         self.data = free_to_loop_index(self.data, n)
         self.data = makepareto(self.data)
 
-    def alloc(self, resource_name: str, size: int, nloops: int):
-        n = nameloop2col(resource_name, nloops)
+    def alloc(self, resource_name: str, size: int, above_loop_index: int):
+        n = nameloop2col(resource_name, above_loop_index)
         if n in self.data:
             self.data[n] += size
         else: