面向多种 AI 芯片的开源、统一编译器 打造多元 AI 芯片编译器及相关工具平台,发展和壮大 Triton 上下游生态
项目当前处于初期,目标是兼容现有适配方案,快速实现单版本多后端支持。
安装依赖(注意使用正确的 python3.x 执行):
- apt install zlib1g zlib1g-dev libxml2 libxml2-dev # ubuntu
- cd python; python3 -m pip install -r requirements.txt
复制代码 编译安装,目前支持的后端 backendxxx 包括 iluvatar、xpu、mthreads、cambricon(有限支持):
- cd python
- export FLAGTREE_BACKEND=backendxxx
- python3 -m pip install . --no-build-isolation -v
复制代码 构建技巧自动下载依赖库的速度可能受限于网络环境,编译前可自行下载至缓存目录 ~/.flagtree(可通过环境变量 FLAGTREE_CACHE_DIR 修改),无需自行设置 LLVM_BUILD_DIR 等环境变量。 运行测试安装完成后可以在后端目录下运行测试: - cd third_party/backendxxx/python/test
- python3 -m pytest -s
复制代码提取码下载:
|