[build] Disable MSCCL++ compilation by default (#1879)
* Enable MSCCLPP on request * Updating docs and README * Updates to CHANGELOG.md * Update CHANGELOG.md Co-authored-by: Jeffrey Novotny <jnovotny@amd.com> * Updates to CHANGELOG.md * Update CHANGELOG.md Co-authored-by: corey-derochie-amd <161367113+corey-derochie-amd@users.noreply.github.com> * Update CHANGELOG.md Github didn't take the edit to my suggestion properly. --------- Co-authored-by: amd <amd@super3.amd.com> Co-authored-by: Jeffrey Novotny <jnovotny@amd.com> Co-authored-by: corey-derochie-amd <161367113+corey-derochie-amd@users.noreply.github.com>
This commit is contained in:
Avinash
GitHub
+6
-6
@@ -26,7 +26,7 @@ install_library=false
|
||||
install_prefix="${ROCM_PATH}"
|
||||
log_trace=false
|
||||
msccl_kernel_enabled=true
|
||||
mscclpp_enabled=true
|
||||
mscclpp_enabled=false
|
||||
enable_mscclpp_clip=false
|
||||
num_parallel_jobs=$(nproc)
|
||||
npkit_enabled=false
|
||||
@@ -51,7 +51,7 @@ function display_help()
|
||||
echo " --enable_backtrace Build with custom backtrace support"
|
||||
echo " --disable-colltrace Build without collective trace"
|
||||
echo " --disable-msccl-kernel Build without MSCCL kernels"
|
||||
echo " --disable-mscclpp Build without MSCCL++ support"
|
||||
echo " --enable-mscclpp Build with MSCCL++ support"
|
||||
echo " --enable-mscclpp-clip Build MSCCL++ with clip wrapper on bfloat16 and half addition routines"
|
||||
echo " --disable-roctx Build without ROCTX logging"
|
||||
echo " -f|--fast Quick-build RCCL (local gpu arch only, no backtrace, and collective trace support)"
|
||||
@@ -82,7 +82,7 @@ function display_help()
|
||||
# check if we have a modern version of getopt that can handle whitespace and long parameters
|
||||
getopt -T
|
||||
if [[ "$?" -eq 4 ]]; then
|
||||
GETOPT_PARSE=$(getopt --name "${0}" --options cdfhij:lprt --longoptions address-sanitizer,dependencies,debug,enable-code-coverage,enable_backtrace,disable-colltrace,disable-msccl-kernel,disable-mscclpp,fast,help,install,jobs:,local_gpu_only,amdgpu_targets:,no_clean,npkit-enable,log-trace,openmp-test-enable,roctx-enable,package_build,prefix:,rm-legacy-include-dir,run_tests_all,run_tests_quick,static,tests_build,time-trace,force-reduce-pipeline,verbose -- "$@")
|
||||
GETOPT_PARSE=$(getopt --name "${0}" --options cdfhij:lprt --longoptions address-sanitizer,dependencies,debug,enable-code-coverage,enable_backtrace,disable-colltrace,disable-msccl-kernel,enable-mscclpp,fast,help,install,jobs:,local_gpu_only,amdgpu_targets:,no_clean,npkit-enable,log-trace,openmp-test-enable,roctx-enable,package_build,prefix:,rm-legacy-include-dir,run_tests_all,run_tests_quick,static,tests_build,time-trace,force-reduce-pipeline,verbose -- "$@")
|
||||
else
|
||||
echo "Need a new version of getopt"
|
||||
exit 1
|
||||
@@ -104,7 +104,7 @@ while true; do
|
||||
--enable_backtrace) build_bfd=true; shift ;;
|
||||
--disable-colltrace) collective_trace=false; shift ;;
|
||||
--disable-msccl-kernel) msccl_kernel_enabled=false; shift ;;
|
||||
--disable-mscclpp) mscclpp_enabled=false; shift ;;
|
||||
--enable-mscclpp) mscclpp_enabled=true; shift ;;
|
||||
--enable-mscclpp-clip) enable_mscclpp_clip=true; shift ;;
|
||||
--disable-roctx) roctx_enabled=false; shift ;;
|
||||
-f | --fast) build_local_gpu_only=true; collective_trace=false; msccl_kernel_enabled=false; shift ;;
|
||||
@@ -247,8 +247,8 @@ if [[ "${msccl_kernel_enabled}" == false ]]; then
|
||||
cmake_common_options="${cmake_common_options} -DENABLE_MSCCL_KERNEL=OFF"
|
||||
fi
|
||||
|
||||
if [[ "${mscclpp_enabled}" == false ]]; then
|
||||
cmake_common_options="${cmake_common_options} -DENABLE_MSCCLPP=OFF"
|
||||
if [[ "${mscclpp_enabled}" == true ]]; then
|
||||
cmake_common_options="${cmake_common_options} -DENABLE_MSCCLPP=ON"
|
||||
fi
|
||||
|
||||
if [[ "${enable_mscclpp_clip}" == true ]]; then
|
||||
|
||||
مرجع در شماره جدید
Block a user