基于LLVM13 Enzyme 安装

一、docker pull 的镜像 ：https://hub.docker.com/r/oneflowinc/llvm13_cuda11.2

二、Enzyme下载 ：https://github.com/EnzymeAD/Enzyme

三、Enzyme官网：https://enzyme.mit.edu/getting_started/UsingEnzyme/

四、安装过程

1、配置LLVM13

bash  #进入workspace
apt-get update
apt-get install vim
vim ~/.bashrc      #进去在最后空白处加上 ： export PATH=$PATH:/usr/lib/llvm-13/bin
source ~/.bashrc
clang -v  #出现版本代码代表成功
1
2
3
4
5
6

2、下载Enzyme

git clone https://github.com/EnzymeAD/Enzyme # 比较慢，也可以保存gitee仓库后，复制gitee仓库保存的Enzyme路径下载
1

3、下载Enzyme所在目录

 root@4e3681f767e1:/workspace/   # 下载后保存的位置
1

4、构建环境

cd /workspace/Enzyme/enzyme  
mkdir build && cd build
cmake -G Ninja .. -DLLVM_DIR=/usr/lib/llvm-13/lib/cmake/llvm -DLLVM_EXTERNAL_LIT=/usr/lib/llvm-13/build/utils/lit/lit.py #可以使用  find /usr -name cmake   和  find /usr -name lit.py  查找这两文件所在位置，一定要匹配 
ninja # 安装过程中的warming 忽略
1
2
3
4

5、跟着官网运行测试，注意路径问题

opt input.ll -load=/path/to/Enzyme/enzyme/build/Enzyme/LLVMEnzyme-<VERSION>.so -enzyme -o output.ll -S # 这个地方需要改改
opt -enable-new-pm=0 input.ll -load=/path/to/Enzyme/enzyme/build/Enzyme/LLVMEnzyme-<VERSION>.so -enzyme -o output.ll -S # 改成这个便不会报错
1
2