Neural Network Based on TVM

Cong Ding's course project of Compiler Design in 2019 Spring, lectured by Prof. Yun (Eric) Liang.

Part 1: Warm-up Project

Use tvm to implement five operators of conventional nerual networks.

Check my implementation in folder "Part1".

The testing framework is published here.

Due: April 24, 2019

Use tvm to optimize two operators: GEMM and Conv2d.

I completed the project with my teammate Jiawei Wu. Check our implementation in folder "Part2".

The testing machine and usage are published here.

Due: June 23, 2019