find c++ ml library to do convolution, and try to put it on vitis

[fann9876543210]

pytorch in C++

[fann9876543210]

FINN

• resnet50 for Alveo U250 only
• https://github.com/Xilinx/finn-examples/tree/main/build

pytorch flow for Vitis AI

• https://github.com/Xilinx/Vitis-AI/tree/master/setup/mpsoc
• https://support.xilinx.com/s/article/Custom-OP-complete-example-design-for-Pytorch?language=en_US

[fann9876543210]

Above site using pre-trained xmodel

---

To train our model on Jupyter notebook

• compile with DPU
• https://github.com/Xilinx/DPU-PYNQ/blob/master/host/train_mnist_model.ipynb