diff --git a/cnn_mapping/resource.py b/cnn_mapping/resource.py index 8200893f6b990a3b33bb187f8f6d10a069369f69..666d351f3b972b0b87ec37e69f05f48b3fc342af 100644 --- a/cnn_mapping/resource.py +++ b/cnn_mapping/resource.py @@ -81,8 +81,8 @@ class Resource(object): assert len(buf_capacity_list) == len(buf_unit_static_cost_list) assert len(buf_capacity_list) == len(para_count_list) - self.bufs = [Buffer(*t) for t in zip(buf_capacity_list, \ - buf_access_cost_list, buf_unit_static_cost_list)] + self.bufs = [Buffer(*t) for t in list(zip(buf_capacity_list, \ + buf_access_cost_list, buf_unit_static_cost_list))] self.num_levels = len(self.bufs) @@ -112,8 +112,8 @@ class Resource(object): # LMEI always assume square-shape array, could change later array_width = [para_count_list[i] if array_dim[i] == 1 else int(math.sqrt(para_count_list[i])) for i in range(self.num_levels)] - self.paras = [Parallelism(*t) for t in zip(para_count_list, \ - partition_mode, array_access_costs, array_dim, array_width)] + self.paras = [Parallelism(*t) for t in list(zip(para_count_list, \ + partition_mode, array_access_costs, array_dim, array_width))] self.access_cost = buf_access_cost_list # If list does not contain 3 separate access costs for (inputs, weights, psum) # assume they all have the same cost