-
This repo provides some speed-up abilites for AI-models in autonomous driving system, which can help these models running fatser on ascend device.
-
The main three parts is as follows:
- common, general modules, which contain some fused pytorch modules of which provide speed-up ability for models runnning on ascend device.
- motion, the motion prediction modules, which are used for trace-planning and motion-prediction. The modules contain some pytorch-custom-apis, the kernels of which are affinitive on ascend device.
- perception, the perception modules, which are used for 3D detection and segmentation. It helps recognize the components on the road.