diff --git a/tests/test_verify_custom_ops.py b/tests/test_verify_custom_ops.py new file mode 100644 index 0000000000000000000000000000000000000000..2f0e654a43ba04b14a1b147f2ce0a699c5653bb2 --- /dev/null +++ b/tests/test_verify_custom_ops.py @@ -0,0 +1,17 @@ +import onnx +from finn.custom_op.layout_custom_ops import CustomOp_Layout + +def test_verify_layout_custom_ops(): + m_node = onnx.helper.make_node( + "MultiThreshold", + ["v", "thresholds"], + ["out"], + domain="finn", + out_scale=2.0, + out_bias=-1.0, + out_dtype="", + ) + + inst = CustomOp_Layout[m_node.op_type] + inst.verify_layout(m_node) +