From 53da3c98563fcf12494b07c1ee83920976160bf7 Mon Sep 17 00:00:00 2001
From: auphelia <jakobapk@web.de>
Date: Mon, 30 Mar 2020 18:20:58 +0100
Subject: [PATCH] [Registry] add data width converter to registry
---
src/finn/custom_op/registry.py | 4 ++++
1 file changed, 4 insertions(+)
diff --git a/src/finn/custom_op/registry.py b/src/finn/custom_op/registry.py
index 0b0d0ee18..c797affff 100644
--- a/src/finn/custom_op/registry.py
+++ b/src/finn/custom_op/registry.py
@@ -39,6 +39,9 @@ from finn.custom_op.multithreshold import MultiThreshold
from finn.custom_op.streamingdataflowpartition import StreamingDataflowPartition
from finn.custom_op.xnorpopcount import XnorPopcountMatMul
from finn.custom_op.maxpoolnhwc import MaxPoolNHWC
+from finn.custom_op.fpgadataflow.streamingdatawidthconverter_batch import (
+ StreamingDataWidthConverter_Batch,
+)
# create a mapping of all known CustomOp names and classes
custom_op = {}
@@ -52,6 +55,7 @@ custom_op["ConvolutionInputGenerator"] = ConvolutionInputGenerator
custom_op["TLastMarker"] = TLastMarker
custom_op["StreamingDataflowPartition"] = StreamingDataflowPartition
custom_op["MaxPoolNHWC"] = MaxPoolNHWC
+custom_op["StreamingDataWidthConverter_Batch"] = StreamingDataWidthConverter_Batch
def getCustomOp(node):
--
GitLab