From 35cdd9e083115df2339a496a016121d97a2320d7 Mon Sep 17 00:00:00 2001 From: auphelia <jakobapk@web.de> Date: Tue, 10 Dec 2019 11:02:08 +0000 Subject: [PATCH] [Execution] Renamed file to verify_custom_op_construct --- .../{layout_custom_ops.py => verify_custom_op_construct.py} | 2 +- tests/test_verify_custom_ops.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) rename src/finn/custom_op/{layout_custom_ops.py => verify_custom_op_construct.py} (98%) 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 bb0a190ad..d2e3556eb 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 2f0e654a4..5d49303b0 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) -- GitLab