diff --git a/src/finn/custom_op/fpgadataflow/__init__.py b/src/finn/custom_op/fpgadataflow/__init__.py index 0160c5f02381b1af55182f168a0df7b202635266..6a37519d372687fbd21fea2c326f55531b6f1eea 100644 --- a/src/finn/custom_op/fpgadataflow/__init__.py +++ b/src/finn/custom_op/fpgadataflow/__init__.py @@ -40,7 +40,11 @@ class HLSCustomOp(CustomOp): self.code_gen_dict = {} def get_nodeattr_types(self): - return {"code_gen_dir": ("s", False, ""), "executable_path": ("s", False, "")} + return { + "backend": ("s", True, "fpgadataflow"), + "code_gen_dir": ("s", False, ""), + "executable_path": ("s", False, ""), + } def code_generation(self, model): node = self.onnx_node diff --git a/src/finn/custom_op/fpgadataflow/streamingfclayer_batch.py b/src/finn/custom_op/fpgadataflow/streamingfclayer_batch.py index 0e64694317e978015bbcc729ed5197d2af3fdfed..975da666f0ba728f542b06865aaa2c66c5f07c07 100644 --- a/src/finn/custom_op/fpgadataflow/streamingfclayer_batch.py +++ b/src/finn/custom_op/fpgadataflow/streamingfclayer_batch.py @@ -14,9 +14,9 @@ class StreamingFCLayer_Batch(HLSCustomOp): def get_nodeattr_types(self): my_attrs = { - "backend": ("s", True, "fpgadataflow"), - "code_gen_dir": ("s", True, ""), - "executable_path": ("s", True, ""), + # "backend": ("s", True, "fpgadataflow"), + # "code_gen_dir": ("s", True, ""), + # "executable_path": ("s", True, ""), "PE": ("i", True, 0), "SIMD": ("i", True, 0), "MW": ("i", True, 0), diff --git a/src/finn/custom_op/fpgadataflow/streamingmaxpool_batch.py b/src/finn/custom_op/fpgadataflow/streamingmaxpool_batch.py index 5e90d37e6de72e0d37fc5c1810056966dfa6250a..92f499b6771efe4455e7259a8eb62ab9c636cb1f 100644 --- a/src/finn/custom_op/fpgadataflow/streamingmaxpool_batch.py +++ b/src/finn/custom_op/fpgadataflow/streamingmaxpool_batch.py @@ -4,9 +4,9 @@ from finn.custom_op.fpgadataflow import HLSCustomOp class StreamingMaxPool_Batch(HLSCustomOp): def get_nodeattr_types(self): my_attrs = { - "backend": ("s", True, "fpgadataflow"), - "code_gen_dir": ("s", True, ""), - "executable_path": ("s", True, ""), + # "backend": ("s", True, "fpgadataflow"), + # "code_gen_dir": ("s", True, ""), + # "executable_path": ("s", True, ""), "ImgDim": ("i", True, 0), "PoolDim": ("i", True, 0), "NumChannels": ("i", True, 0),