Skip to content
Snippets Groups Projects
Commit 3f260f6b authored by auphelia's avatar auphelia
Browse files

[member function] Changed get_nodeattr_types by moving "backend" attribute to...

[member function] Changed get_nodeattr_types by moving "backend" attribute to __init__ instead of in each fpgadataflow custom op class
parent e75df48c
No related branches found
No related tags found
No related merge requests found
......@@ -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
......
......@@ -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),
......
......@@ -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),
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment