Enable kernarg preloading for ROCm 6.1 (#1068)

Signed-off-by: nileshnegi <Nilesh.Negi@amd.com>
This commit is contained in:
Nilesh M Negi
2024-02-01 12:14:04 -06:00
کامیت شده توسط GitHub
والد 01b359027b
کامیت 2458f158b1
+1 -1
مشاهده پرونده
@@ -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)