编译运行
介绍目录结构
样例代码目录结构如下所示:
1 2 3 4 5 6 | CommInitAll ├── bin │ └── CommInitTest ├── Makefile └── src └── main.cc |
- Makefile文件内容可参见配置Makefile文件。
- main.cc的样例代码可参见样例代码。
- bin目录下的CommInitTest为编译后生成的可执行文件。
配置环境变量
配置样例编译时依赖的环境变量:
1 2 3 | source /usr/local/Ascend/ascend-toolkit/set_env.sh export PATH=/usr/local/mpich-3.2.1/bin:$PATH export LD_LIBRARY_PATH=/usr/local/mpich-3.2.1/lib:$LD_LIBRARY_PATH |
- “/usr/local/Ascend”为CANN软件root用户的默认安装路径,如果使用普通用户安装,或指定路径安装,请自行替换。
- “/usr/local/mpich-3.2.1/lib”为安装后MPI的库文件所在路径,请根据实际情况替换。
配置Makefile文件
Makefile文件示例如下,请根据实际情况替换源文件及其所在路径。
#
#loading path
#--------------------------------------------------------------------------------------------------------------------------------------------------
CXXFLAGS := -std=c++11\
-Werror\
-fstack-protector-strong\
-fPIE -pie\
-O2\
-s\
-Wl,-z,relro\
-Wl,-z,now\
-Wl,-z,noexecstack\
-Wl,--copy-dt-needed-entries
#path of the source file
Common_DIR = ./src
#source file
Common_SRC = $(wildcard ${Common_DIR}/*.cc)
HCCL_INC_DIR = ${ASCEND_DIR}/include
HCCL_LIB_DIR = ${ASCEND_DIR}/lib64
ACL_INC_DIR = ${ASCEND_DIR}/include
ACL_LIB_DIR = ${ASCEND_DIR}/lib64
MPI_INC_DIR = ${MPI_HOME}/include
MPI_LIB_DIR = ${MPI_HOME}/lib
LIST = CommInitTest
#
#library flags
#--------------------------------------------------------------------------------------------------------------------------------------------------
LIBS = -L$(HCCL_LIB_DIR) -lhccl\
-L$(ACL_LIB_DIR) -lascendcl\
-L$(MPI_LIB_DIR) -lmpi
INCLUDEDIRS = -I$(HCCL_INC_DIR)\
-I$(ACL_INC_DIR)\
-I$(MPI_INC_DIR)\
-I$(Common_DIR)
#
#make
#--------------------------------------------------------------------------------------------------------------------------------------------------
all:
@mkdir -p bin
g++ $(CXXFLAGS) $(Common_SRC) $(INCLUDEDIRS) -o CommInitTest $(LIBS)
@printf "\nCommInitTest compile completed\n"
mv $(LIST) ./bin
.PHONY: clean
clean:
rm -rf ./bin/*Test
编译样例
执行如下命令编译样例:
1 | make MPI_HOME=/usr/local/mpich-3.2.1/ ASCEND_DIR=/usr/local/Ascend/ascend-toolkit/latest |
编译成功后,会在执行编译命令的bin/目录下生成CommInitTest可执行文件。
执行样例
编译完成后,通过以下指令启动:
1 | mpirun -n 1 ./bin/CommInitTest |
HcclCommInitAll初始化方式为单进程多线程的拉起方式,所以该初始化方式的样例执行命令为mpirun -n 1 ./bin/CommInitTest。
结果示例
- HcclSend/HcclRecv代码样例中,接收数据的rank ID分别为4、5、6、7,每个rank分别接收8个数据(i0 ~ i8),数据为“tmpResBuff”的值“2”,输出结果示例如下:
rankId:4,i0 2 rankId:4,i1 2 rankId:4,i2 2 rankId:4,i3 2 rankId:4,i4 2 rankId:4,i5 2 rankId:4,i6 2 rankId:4,i7 2 ... rankId:7,i4 2 rankId:7,i5 2 rankId:7,i6 2 rankId:7,i7 2
- HcclAllReduce代码样例中,共8个rank,每个rank中有8个数据,每个数据的初始值为2,执行HCCL_REDUCE_SUM操作后,每个数据的值为16,输出结果打印出了每个rank的每个值,结果示例如下:
rankId:0,i0 16 rankId:0,i1 16 rankId:0,i2 16 ... rankId:7,i5 16 rankId:7,i6 16 rankId:7,i7 16
父主题: HcclCommInitAll初始化方式