From 347538ae74522a98359aeb0e659e41644b8926aa Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Wed, 22 Apr 2020 23:17:12 +0100
Subject: [PATCH] [Util] add get_num_default_workers to access env.var.
---
src/finn/util/basic.py | 11 +++++++++++
1 file changed, 11 insertions(+)
diff --git a/src/finn/util/basic.py b/src/finn/util/basic.py
index f99a453d0..4eb0e6cb8 100644
--- a/src/finn/util/basic.py
+++ b/src/finn/util/basic.py
@@ -44,6 +44,17 @@ pynq_part_map["Pynq-Z2"] = "xc7z020clg400-1"
pynq_part_map["ZCU104"] = "xczu7ev-ffvc1156-2-e"
+def get_num_default_workers():
+ """Return the number of workers for parallel transformations. Controllable
+ via the NUM_DEFAULT_WORKERS environment variable. If the env.var. is
+ undefined, the default value of 1 is returned.
+ """
+
+ try:
+ return int(os.environ["NUM_DEFAULT_WORKERS"])
+ except KeyError:
+ return 1
+
def get_finn_root():
"Return the root directory that FINN is cloned into."
--
GitLab