From 701f1ac6c3844845c7270b73b3bb23f4bb0571b7 Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Wed, 20 Nov 2019 16:03:08 +0000
Subject: [PATCH] [Wrapper] add get_hls_datatype_str
---
src/finn/core/datatype.py | 10 ++++++++++
1 file changed, 10 insertions(+)
diff --git a/src/finn/core/datatype.py b/src/finn/core/datatype.py
index 42a366aaf..34eb5abc2 100644
--- a/src/finn/core/datatype.py
+++ b/src/finn/core/datatype.py
@@ -130,3 +130,13 @@ class DataType(Enum):
"""Return whether this DataType represents integer values only."""
# only FLOAT32 is noninteger for now
return self != DataType.FLOAT32
+
+ def get_hls_datatype_str(self):
+ """Return the corresponding Vivado HLS datatype name."""
+ if self.is_integer():
+ if self.signed():
+ return "ap_int<%d>" % self.bitwidth()
+ else:
+ return "ap_uint<%d>" % self.bitwidth()
+ else:
+ return "float"
--
GitLab