MetaX TVM: Open Deep Learning Compiler Stack

English | 中文版

简介

Apache TVM是发展最早和最成功的AI编译器之一，支持从主流的深度学习框架比如pytorch导入模型，在优化编译到多种硬件后端，有着非常活跃的社区讨论和完善的文档支持。

本项目（mcTVM）在 TVM v0.18.0 release 的基础上，增加了对沐曦（MetaX）GPU的支持。

快速开始

源码编译

依赖：

• 沐曦MACA(MetaX Advanced Compute Architecture)编程环境，参见沐曦开发者社区的《曦云系列_通用计算GPU_快速上手指南》
• 其他环境要求请参考TVM社区文档

git clone https://github.com/MetaX-MACA/mcTVM.git mcTVM
cd mcTVM
git submodule update --init --recursive
rm -rf build && mkdir build && cd build
cp ../cmake/config.cmake ./
# USE_MACA 已经默认打开
cmake ../ && make -j $(nproc)
export TVM_HOME=/path-to-mcTVM
export PYTHONPATH=$TVM_HOME/python:$PYTHONPATH

使用MACA后端

import tvm
# define maca target by kind name
target = tvm.target.Target("maca")
# or specific target by tag name
target = tvm.target.Target("metax/mxc-c500")
# define maca device
dev = tvm.maca()
# or
dev = tvm.device("maca")

贡献

参见 CONTRIBUTING_ZH.md