diff --git a/src/finn/data/onnx/finn-hls-model/tfc_w1_a1_after_conv_to_hls.onnx b/src/finn/data/onnx/finn-hls-model/tfc_w1_a1_after_conv_to_hls.onnx new file mode 100644 index 0000000000000000000000000000000000000000..aada6f07e9d3910122d2eb357d8a8c1224e9fbab Binary files /dev/null and b/src/finn/data/onnx/finn-hls-model/tfc_w1_a1_after_conv_to_hls.onnx differ diff --git a/tests/fpgadataflow/test_create_dataflow_partition.py b/tests/fpgadataflow/test_create_dataflow_partition.py new file mode 100644 index 0000000000000000000000000000000000000000..e0b8b491b0ed926a56331a8e125f3de4ecd91615 --- /dev/null +++ b/tests/fpgadataflow/test_create_dataflow_partition.py @@ -0,0 +1,21 @@ +import os.path +from pkgutil import get_data + +from finn.core.modelwrapper import ModelWrapper +from finn.custom_op.registry import getCustomOp +from finn.transformation.fpgadataflow.create_dataflow_partition import ( + CreateDataflowPartition, +) + + +def test_create_dataflow_partition(): + # load the onnx model + raw_m = get_data( + "finn", "data/onnx/finn-hls-model/tfc_w1_a1_after_conv_to_hls.onnx" + ) + model = ModelWrapper(raw_m) + model = model.transform(CreateDataflowPartition()) + assert model.graph.node[2].op_type == "StreamingDataflowPartition" + sdp_node = getCustomOp(model.graph.node[2]) + assert sdp_node.__class__.__name__ == "StreamingDataflowPartition" + assert os.path.isfile(sdp_node.get_nodeattr("model"))