Update MACA

repos 2024-11-06 10:35:05 +08:00
parent bd9b35f410
commit a3c150762a

22
MACA.md

@ -372,6 +372,28 @@
conda install triton=<version>=<build> -c https://repos.metax-tech.com/r/maca-conda/ conda install triton=<version>=<build> -c https://repos.metax-tech.com/r/maca-conda/
``` ```
## 验证安装
- **运行前设置环境变量**
```bash
export MACA_PATH=/opt/maca/
export LD_LIBRARY_PATH=${MACA_PATH}/lib:${MACA_PATH}/mxgpu_llvm/lib:${LD_LIBRARY_PATH}
export MACA_CLANG_PATH=${MACA_PATH}/mxgpu_llvm/bin
export PYTORCH_USE_FLASHATTN=1
```
- **如果使用PyTorch Extension功能需要安装cu-bridge环境并额外设置以下环境变量**
```bash
export CUDA_PATH=/opt/maca/tools/cu-bridge/
export CUCC_PATH=/opt/maca/tools/cu-bridge/
export PATH=${CUCC_PATH}/tools:${CUCC_PATH}/bin:${PATH}
```
- ** 执行以下命令会输出 tensor([1., 1.], device='cuda:0') 即表明安装成功**
```bash
python -c "import torch; print(torch.ones(2).cuda())"
```
## 需要注意 ## 需要注意
### 包依赖 ### 包依赖