Enable kernarg preloading for ROCm 6.1 (#1068)
Signed-off-by: nileshnegi <Nilesh.Negi@amd.com>
[ROCm/rccl commit: 2458f158b1]
Šī revīzija ir iekļauta:
revīziju iesūtīja
GitHub
vecāks
96479caeda
revīzija
f23716de80
@@ -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)
|
||||
|
||||
Atsaukties uz šo jaunā problēmā
Block a user