Skip to content

Commit 5877121

Browse files
authored
Merge pull request #1058 from InfiniTensor/issue/1033d
issue/1033 - update doc
2 parents f3f4bf1 + c1ac5a6 commit 5877121

1 file changed

Lines changed: 9 additions & 4 deletions

File tree

README.md

Lines changed: 9 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -159,14 +159,19 @@ python scripts/install.py [XMAKE_CONFIG_FLAGS]
159159

160160
```shell
161161
162-
(1) 在third_party目录拉取cutlass和flash attn库的源码(不需要--recursive)
162+
# 在third_party目录拉取cutlass和flash attn库的源码(不需要--recursive)
163163
164-
(2) 设置(1)中cutlass路径的环境变量CUTLASS_ROOT
164+
# 设置cutlass路径的环境变量CUTLASS_ROOT(部分环境可选)
165+
export CUTLASS_ROOT=<path-to>/InfiniCore/third_party/cutlass
165166
166-
(3) xmake配置环节额外打开 --aten 开关,并设置 --flash-attn 库位置,例:
167+
# xmake配置环节额外打开 --aten 开关,并设置 --flash-attn 库位置,例(cuda路径部分环境可使用默认)
167168
xmake f --nv-gpu=y --ccl=y --cuda=$CUDA_HOME --aten=y --flash-attn=<path-to>/InfiniCore/third_party/flash-attention -cv
168169
169-
(4) flash attenion库会伴随infinicore_cpp_api一同编译安装
170+
# 设置额外的环境变量
171+
export CPLUS_INCLUDE_PATH=$CUDA_HOME/include:$CPLUS_INCLUDE_PATH
172+
173+
# flash attenion库会伴随infinicore_cpp_api一同编译安装
174+
170175
```
171176

172177
2. 编译安装

0 commit comments

Comments
 (0)