File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff 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
1721772. 编译安装
You can’t perform that action at this time.
0 commit comments