From 73087901b1899b6e89f8a51b1e5d2087cc9b0bb4 Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <yamanu@xilinx.com>
Date: Fri, 10 Jul 2020 15:54:52 +0100
Subject: [PATCH] [Transform] call InferDataTypes in
MoveScalarLinearPastInvariants
---
src/finn/transformation/streamline/reorder.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/src/finn/transformation/streamline/reorder.py b/src/finn/transformation/streamline/reorder.py
index 2cde19143..452366607 100644
--- a/src/finn/transformation/streamline/reorder.py
+++ b/src/finn/transformation/streamline/reorder.py
@@ -560,7 +560,9 @@ class MoveScalarLinearPastInvariants(Transformation):
graph_modified = True
else:
continue
- model = model.transform(InferShapes())
+ if graph_modified:
+ model = model.transform(InferShapes())
+ model = model.transform(InferDataTypes())
return (model, graph_modified)
--
GitLab