From 6b0672af60c918b833383f3f71de599e51b18fbf Mon Sep 17 00:00:00 2001
From: bmccolm <briannamccolm@gmail.com>
Date: Mon, 15 Jan 2024 22:01:11 -0800
Subject: [PATCH] added index to parallel count list
---
tools/run_optimizer.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/tools/run_optimizer.py b/tools/run_optimizer.py
index 31fd40b..8f64d04 100644
--- a/tools/run_optimizer.py
+++ b/tools/run_optimizer.py
@@ -79,7 +79,7 @@ def mac_explore_optimizer(arch_info, network_info, schedule_info, verbose=False)
def dataflow_explore_optimizer(arch_info, network_info, file_name, verbose=False):
- assert arch_info["parallel_count"] > 1, \
+ assert arch_info["parallel_count"][0] > 1, \
"parallel count has to be more than 1 for dataflow exploration"
resource = cm.Resource.arch(arch_info)
--
GitLab