Enable kernarg preloading for ROCm 6.1 (#1068)

Signed-off-by: nileshnegi <Nilesh.Negi@amd.com> [ROCm/rccl commit: 2458f158b1]
2024-02-01 12:14:04 -06:00
parent 96479caeda
commit f23716de80
1 changed files with 1 additions and 1 deletions
@@ -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)