Roadmap

This document lists general directions of TPU-MLIR, include principal goals, plans, etc.

Principal goals

• keep regular sync with mlir from project llvm-project.
• continue maintaining high-quality, well-tested and documented modules.
• support various neuron network modes, such as yolo, ssd, bert, etc.
• support various chips, especially SOPHGO TPU, such as BM1684x, BM1684, CV183x, etc.
• keep high performance and hign accuraccy, especially for INT8 quantization models.
• support various tools to improve development conveniently and efficiently.

Plans

gantt
dateformat YYYY-MM
title Plans in 2022 and 2023

section TPU Support
1684x support : done, crit, task_chip, 2022-03, 2022-06
1684 support : active, crit, task_chip, 2022-10, 2022-12
18xx support : active, crit, task_chip, 2022-10, 2022-12
Athena2 support : active, crit, task_chip, 2022-11, 2022-12

section NN Framework Support
ONNX Support : done, crit, task_NN, 2022-03, 2022-06
TFlite support : done, crit, task_NN, 2022-06, 2022-10
Caffe support : done, crit, task_NN, 2022-09, 2022-10

section Tools
model_runner.py: done, crit, task_tool, 2022-03, 2022-04
npz_tool.py: done, crit, task_tool, 2022-03, 2022-04
run_calibration.py: done, crit, task_tool, 2022-03, 2022-04
bmodel_dis.py: done, crit, task_tool, 2022-06, 2022-09
mlir2onnx.py: active, crit, task_tool, 2022-10, 2023-01

section Functions
layergroup basic: done, crit, task_func, 2022-05, 2022-06
layergroup refine1: done, crit, task_func, 2022-08, 2022-10
layergroup refine2: active, crit, task_func, 2022-10, 2023-01
cpu layer: active, crit, task_func, 2022-10, 2023-01
dynamic compile: active, crit, task_func, 2022-10, 2023-03
mix precision: active, crit, task_func, 2022-11, 2023-03
min/max refine: active, crit, task_func, 2022-12, 2023-03

section NN Models Support
Resnet50: done, crit, task_model, 2022-03, 2022-06
Modilenet: done, crit, task_model, 2022-06, 2022-07
SSD: done, crit, task_model, 2022-06, 2022-07
Yolov5: done, crit, task_model, 2022-06, 2022-07
Vgg16: done, crit, task_model, 2022-07, 2022-08
Shufflenet: done, crit, task_model, 2022-07, 2022-08
Squeezenet: done, crit, task_model, 2022-07,2022-08
Efficientnet: done, crit, task_model, 2022-07,2022-08
Mnist: done, crit, task_model, 2022-07, 2022-08
Densenet: done, crit, task_model, 2022-08, 2022-09
Senet: done, crit, task_model, 2022-08, 2022-09
Inception: done, crit, task_model, 2022-08, 2022-09
Yolov3: done, crit, task_model, 2022-08, 2022-09
Resnet50_tflite: done, crit, task_model, 2022-06, 2022-08
Modilenet_tflite: done, crit, task_model, 2022-08, 2022-10
Inception_tflite: active, crit, task_model, 2022-09, 2022-11
Yolov5_tflite: active, crit, task_model, 2022-11, 2022-12
ssd_mobilenet: active, crit, task_model, 2022-12, 2023-01
mobilebert: active, crit, task_model, 2022-12, 2023-01

Loading