diff --git a/pytimeloop/fastfusion/pareto.py b/pytimeloop/fastfusion/pareto.py index 68f13d3..e5b7fd4 100644 --- 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: