diff --git a/src/finn/custom_op/layout_custom_ops.py b/src/finn/custom_op/verify_custom_op_construct.py similarity index 98% rename from src/finn/custom_op/layout_custom_ops.py rename to src/finn/custom_op/verify_custom_op_construct.py index bb0a190ad76d1379c3f7b14a4117db0e57175b45..d2e3556ebb8346f0e8c19957fa2becb04b550ef2 100644 --- a/src/finn/custom_op/layout_custom_ops.py +++ b/src/finn/custom_op/verify_custom_op_construct.py @@ -1,7 +1,7 @@ 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() diff --git a/tests/test_verify_custom_ops.py b/tests/test_verify_custom_ops.py index 2f0e654a43ba04b14a1b147f2ce0a699c5653bb2..5d49303b0c4073040d029f8e003d5646f31ec1e1 100644 --- a/tests/test_verify_custom_ops.py +++ b/tests/test_verify_custom_ops.py @@ -1,5 +1,5 @@ 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)