From 2ac47b43e2ee0e7f1705262f4389412382e33704 Mon Sep 17 00:00:00 2001
From: bmccolm <briannamccolm@gmail.com>
Date: Mon, 15 Jan 2024 09:34:33 -0800
Subject: [PATCH] changed mapping point syntax to python3
---
cnn_mapping/mapping_point_generator.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/cnn_mapping/mapping_point_generator.py b/cnn_mapping/mapping_point_generator.py
index aa39396..e8243f5 100644
--- a/cnn_mapping/mapping_point_generator.py
+++ b/cnn_mapping/mapping_point_generator.py
@@ -440,7 +440,7 @@ def current_level_partition_blocking_2d_with_hint(loop_tiles, slb, para_count, l
for slps in itertools.product(*[para_perm_1d0, para_perm_1d1]):
slp0, slp1 = slps
- para_index0, para_index1 = para_index_generator.next()
+ para_index0, para_index1 = next(para_index_generator)
if set(para_index0[0]).isdisjoint(set(para_index1[0])):
combined_slp = [a * b for a, b in list(zip(slp0, slp1))]
para_permutation.append(combined_slp)
@@ -460,7 +460,7 @@ def current_level_partition_blocking_2d(loop_tiles, slb, para_count, layer, u_th
for slps in itertools.combinations(para_perm_1d, 2):
slp0, slp1 = slps
- para_index0, para_index1 = para_index_generator.next()
+ para_index0, para_index1 = next(para_index_generator)
if set(para_index0[0]).isdisjoint(set(para_index1[0])):
combined_slp = [a * b for a, b in list(zip(slp0, slp1))]
para_permutation.append(combined_slp)
--
GitLab