Skip to content
Snippets Groups Projects
Commit 99d5c84b authored by Yaman Umuroglu's avatar Yaman Umuroglu
Browse files

[Test] more progress towards cnv-w1a1 hls conversion

parent 51df35f6
No related branches found
No related tags found
No related merge requests found
......@@ -38,6 +38,7 @@ import torch
import finn.core.onnx_exec as oxe
import finn.transformation.fpgadataflow.convert_to_hls_layers as to_hls
import finn.transformation.streamline.absorb as absorb
from finn.transformation.streamline.reorder import MakeMaxPoolNHWC
from finn.core.modelwrapper import ModelWrapper
from finn.custom_op.registry import getCustomOp
from finn.transformation.bipolar_to_xnor import ConvertBipolarMatMulToXnorPopcount
......@@ -138,6 +139,8 @@ def test_convert_to_hls_layers_cnv_w1a1():
model = model.transform(Streamline())
model.save("cnv-streamline.onnx")
model = model.transform(LowerConvsToMatMul())
model = model.transform(MakeMaxPoolNHWC())
model = model.transform(absorb.AbsorbTransposeIntoMultiThreshold())
model.save("cnv-lower.onnx")
......
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