From 6933d080379a9499f4200e6658ba38291b9c54f2 Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <yamanu@amd.com>
Date: Thu, 22 Sep 2022 14:44:37 +0200
Subject: [PATCH] [Refactor] add missing ind param to Checksum HLSCustomOp

---
 src/finn/custom_op/fpgadataflow/checksum.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/finn/custom_op/fpgadataflow/checksum.py b/src/finn/custom_op/fpgadataflow/checksum.py
index 21a09a1c5..c927c07df 100644
--- a/src/finn/custom_op/fpgadataflow/checksum.py
+++ b/src/finn/custom_op/fpgadataflow/checksum.py
@@ -98,7 +98,7 @@ class CheckSum(HLSCustomOp):
     def get_folded_input_shape(self, ind=0):
         return self.get_nodeattr("folded_shape")
 
-    def get_folded_output_shape(self):
+    def get_folded_output_shape(self, ind=0):
         return self.get_nodeattr("folded_shape")
 
     def get_normal_input_shape(self, ind=0):
-- 
GitLab