yezhengmao/nccl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

address review comments

Browse Source
This commit is contained in:
Jane Xu 2022-09-14 11:16:17 -04:00 committed by Sylvain Jeaugey
parent 79fb0326ac
commit f89fd4777d
1 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
src/collectives/device/gen_rules.sh
View File

@ -13,9 +13,6 @@ then
datatypes+=" bf16" datatypes+=" bf16"
fi fi
echo "CURDIR := \$(dir \$(realpath \$(word \$(words \$(math\$(MAKEFILE_LIST))-1), \$(MAKEFILE_LIST))))"
+echo ""
targets="GENOBJS := \\\\\n" targets="GENOBJS := \\\\\n"
for base in sendrecv all_reduce all_gather broadcast reduce reduce_scatter; do for base in sendrecv all_reduce all_gather broadcast reduce reduce_scatter; do
@ -26,11 +23,12 @@ for base in sendrecv all_reduce all_gather broadcast reduce reduce_scatter; do
for dt in ${datatypes}; do for dt in ${datatypes}; do
# Generate a unique filename for each compilation unit, # Generate a unique filename for each compilation unit,
# otherwise the __nv_module_id may conflict at link time # otherwise the __nv_module_id may conflict at link time
echo "${dir}/${base}_${opn}_${dtn}.cu :" echo "${dir}/${base}_${op}_${dt}.cu : ${base}.cu"
echo " echo \"#include \\\"\$(CURDIR)${base}.cu\\\"\" > \$@" echo " @printf \"Copying %-35s > %s\\\\n\" \$< \$@"
echo " cp \$< \$@"
echo "" echo ""
# Compile the file # Compile the file
echo "${dir}/${base}_${op}_${dt}.o : ${dir}/${base}_${opn}_${dtn}.cu ${base}.cu ${dir}/${base}.dep" echo "${dir}/${base}_${op}_${dt}.o : ${dir}/${base}_${op}_${dt}.cu ${base}.cu ${dir}/${base}.dep"
echo " @printf \"Compiling %-35s > %s\\\\n\" ${base}.cu ${dir}/${base}_${op}_${dt}.o" echo " @printf \"Compiling %-35s > %s\\\\n\" ${base}.cu ${dir}/${base}_${op}_${dt}.o"
echo " mkdir -p ${dir}" echo " mkdir -p ${dir}"