From a3c150762af65f95a1ddca71494b2cd74e59776c Mon Sep 17 00:00:00 2001
From: repos <lyni@metax-tech.com>
Date: Wed, 6 Nov 2024 10:35:05 +0800
Subject: [PATCH] Update MACA

---
 MACA.md | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

diff --git a/MACA.md b/MACA.md
index 778e620..a7b6059 100644
--- a/MACA.md
+++ b/MACA.md
@@ -372,6 +372,28 @@
       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())"
+   ```
+
+
 ## 需要注意
 
 ### 包依赖