From 5e33a8f66cea5f2f36383ee59b4ca9fff6e290ae Mon Sep 17 00:00:00 2001
From: Tobi-Alonso <tobi.alonso@gmail.com>
Date: Thu, 25 Jun 2020 17:36:01 +0100
Subject: [PATCH] Fix comment and infer_node_datatype for cases when in data
type != out data type
---
src/finn/custom_op/fpgadataflow/pool_batch.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/finn/custom_op/fpgadataflow/pool_batch.py b/src/finn/custom_op/fpgadataflow/pool_batch.py
index 1aedf7056..c7edc24d0 100644
--- a/src/finn/custom_op/fpgadataflow/pool_batch.py
+++ b/src/finn/custom_op/fpgadataflow/pool_batch.py
@@ -45,7 +45,7 @@ class Pool_Batch(HLSCustomOp):
# note: the actual data layout produced by the hlslib kernels is different
# for depthwise ops.
- # * depthwise SWG: (1, OFMDim, OFMDim, IFMChannels/SIMD, K, K, SIMD)
+ # * depthwise SWG: (1, OFMDim, OFMDim, IFMChannels/PE, K, K, PE)
Channels can be folded using PE (SIMD from the input perspective)
TODO: doc
@@ -159,7 +159,7 @@ class Pool_Batch(HLSCustomOp):
def infer_node_datatype(self, model):
node = self.onnx_node
# data type stays the same
- dtype = model.get_tensor_datatype(node.input[0])
+ dtype = self.get_output_datatype()
model.set_tensor_datatype(node.output[0], dtype)
def verify_node(self):
--
GitLab