Skip to content
Snippets Groups Projects
Commit 35cdd9e0 authored by auphelia's avatar auphelia
Browse files

[Execution] Renamed file to verify_custom_op_construct

parent ff6f1860
No related branches found
No related tags found
No related merge requests found
from enum import Enum, auto
from finn.core.utils import get_by_name
class CustomOp_Layout(Enum):
class CustomOp_Construct(Enum):
MultiThreshold = auto()
#XnorPopcountMatMul = auto()
#StreamingMaxPool_Batch = auto()
......
import onnx
from finn.custom_op.layout_custom_ops import CustomOp_Layout
from finn.custom_op.verify_custom_op_construct import CustomOp_Construct
def test_verify_layout_custom_ops():
m_node = onnx.helper.make_node(
......@@ -12,6 +12,6 @@ def test_verify_layout_custom_ops():
out_dtype="",
)
inst = CustomOp_Layout[m_node.op_type]
inst = CustomOp_Construct[m_node.op_type]
inst.verify_layout(m_node)
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