diff --git a/projects/rccl/CMakeLists.txt b/projects/rccl/CMakeLists.txt index 0ac1ed733e..9f8b48ba42 100644 --- a/projects/rccl/CMakeLists.txt +++ b/projects/rccl/CMakeLists.txt @@ -164,6 +164,12 @@ if("${HIP_COMPILER}" MATCHES "clang") endforeach() target_link_libraries(rccl PRIVATE -fgpu-rdc) target_include_directories(rccl PRIVATE /opt/rocm/hsa/include) + find_program( hipcc_executable hipcc ) + execute_process(COMMAND bash "-c" "${hipcc_executable} -help | grep 'parallel-jobs'" OUTPUT_VARIABLE hipcc_parallel_jobs) + if("${hipcc_parallel_jobs}" MATCHES "parallel-jobs") + target_compile_options(rccl PRIVATE -parallel-jobs=4 PRIVATE -Wno-format-nonliteral) + target_link_libraries(rccl PRIVATE -parallel-jobs=4) + endif() endif() if("${HIP_COMPILER}" MATCHES "hcc")