From f23716de80d530fa04b2eb61d9e5bc1fbbaf9e48 Mon Sep 17 00:00:00 2001 From: Nilesh M Negi Date: Thu, 1 Feb 2024 12:14:04 -0600 Subject: [PATCH] Enable kernarg preloading for ROCm 6.1 (#1068) Signed-off-by: nileshnegi [ROCm/rccl commit: 2458f158b11a8a972173798d0880cfbb785ac253] --- projects/rccl/CMakeLists.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/projects/rccl/CMakeLists.txt b/projects/rccl/CMakeLists.txt index abf4da4133..12efe8fc8f 100644 --- a/projects/rccl/CMakeLists.txt +++ b/projects/rccl/CMakeLists.txt @@ -635,7 +635,7 @@ else() message(STATUS "Building shared RCCL library") endif() if (HAVE_KERNARG_PRELOAD) - target_link_options(rccl PRIVATE -mllvm --amdgpu-kernarg-preload-count=16) + target_link_options(rccl PRIVATE -Xoffload-linker -mllvm=-amdgpu-kernarg-preload-count=16) endif() target_link_libraries(rccl PRIVATE Threads::Threads)