Skip to content

Commit d74270e

Browse files
pruthvistonyfacebook-github-bot
authored andcommitted
Updates to remove HIP deprecated symbols (#355)
Summary: - Removed usage of HIP_HCC_FLAGS - Proper of include path for rccl.h Pull Request resolved: #355 Reviewed By: ezyang Differential Revision: D44859009 Pulled By: malfet fbshipit-source-id: cb476dcafae5110d62697047496abcb742fc381b
1 parent 56b221c commit d74270e

File tree

3 files changed

+10
-10
lines changed

3 files changed

+10
-10
lines changed

cmake/Dependencies.cmake

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -155,13 +155,13 @@ if(USE_ROCM)
155155
list(APPEND HIP_CXX_FLAGS -Wno-duplicate-decl-specifier)
156156
list(APPEND HIP_CXX_FLAGS -DUSE_MIOPEN)
157157

158-
set(HIP_HCC_FLAGS ${HIP_CXX_FLAGS})
158+
set(HIP_CLANG_FLAGS ${HIP_CXX_FLAGS})
159159
# Ask hcc to generate device code during compilation so we can use
160160
# host linker to link.
161-
list(APPEND HIP_HCC_FLAGS -fno-gpu-rdc)
162-
list(APPEND HIP_HCC_FLAGS -Wno-defaulted-function-deleted)
161+
list(APPEND HIP_CLANG_FLAGS -fno-gpu-rdc)
162+
list(APPEND HIP_CLANG_FLAGS -Wno-defaulted-function-deleted)
163163
foreach(gloo_rocm_arch ${GLOO_ROCM_ARCH})
164-
list(APPEND HIP_HCC_FLAGS --amdgpu-target=${gloo_rocm_arch})
164+
list(APPEND HIP_CLANG_FLAGS --amdgpu-target=${gloo_rocm_arch})
165165
endforeach()
166166

167167
set(GLOO_HIP_INCLUDE ${hip_INCLUDE_DIRS} $<BUILD_INTERFACE:${HIPIFY_OUTPUT_ROOT_DIR}> $<INSTALL_INTERFACE:include> ${GLOO_HIP_INCLUDE})

cmake/Modules/Findrccl.cmake

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ else()
2121
endif()
2222

2323
find_path(RCCL_INCLUDE_DIR
24-
NAMES rccl.h
24+
NAMES rccl/rccl.h
2525
HINTS
2626
${RCCL_INCLUDE_DIR}
2727
${RCCL_ROOT_DIR}/include)
@@ -33,17 +33,17 @@ else()
3333
set(RCCL_LIBNAME "rccl")
3434
endif()
3535

36-
find_library(RCCL_LIBRARY
36+
find_library(RCCL_LIB_PATH
3737
NAMES ${RCCL_LIBNAME}
3838
HINTS
3939
${RCCL_LIB_DIR}
4040
${RCCL_ROOT_DIR}/lib)
4141

4242
include(FindPackageHandleStandardArgs)
43-
find_package_handle_standard_args(RCCL DEFAULT_MSG RCCL_INCLUDE_DIR RCCL_LIBRARY)
43+
find_package_handle_standard_args(RCCL DEFAULT_MSG RCCL_INCLUDE_DIR RCCL_LIB_PATH)
4444

4545
if (RCCL_FOUND)
46-
set(RCCL_HEADER_FILE "${RCCL_INCLUDE_DIR}/rccl.h")
46+
set(RCCL_HEADER_FILE "${RCCL_INCLUDE_DIR}/rccl/rccl.h")
4747
message(STATUS "Determining RCCL version from the header file: ${RCCL_HEADER_FILE}")
4848
file (STRINGS ${RCCL_HEADER_FILE} RCCL_MAJOR_VERSION_DEFINED
4949
REGEX "^[ \t]*#define[ \t]+RCCL_MAJOR[ \t]+[0-9]+.*$" LIMIT_COUNT 1)
@@ -53,7 +53,7 @@ if (RCCL_FOUND)
5353
message(STATUS "RCCL_MAJOR_VERSION: ${RCCL_MAJOR_VERSION}")
5454
endif()
5555
set(RCCL_INCLUDE_DIRS ${RCCL_INCLUDE_DIR})
56-
set(RCCL_LIBRARIES ${RCCL_LIBRARY})
56+
set(RCCL_LIBRARIES ${RCCL_LIB_PATH})
5757
message(STATUS "Found RCCL (include: ${RCCL_INCLUDE_DIRS}, library: ${RCCL_LIBRARIES})")
5858
mark_as_advanced(RCCL_ROOT_DIR RCCL_INCLUDE_DIRS RCCL_LIBRARIES)
5959
endif()

tools/amd_build/pyHIPIFY/cuda_to_hip_mappings.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7784,7 +7784,7 @@
77847784
("cudnn", ("miopen", API_GLOO)),
77857785
("namespace cuda", ("namespace hip", API_GLOO)),
77867786
("gloo/cuda.h", ("gloo/hip.h", API_GLOO)),
7787-
("<nccl.h>", ("<rccl.h>", API_GLOO)),
7787+
("<nccl.h>", ("<rccl/rccl.h>", API_GLOO)),
77887788
("GLOO_USE_NCCL", ("GLOO_USE_RCCL", API_GLOO)),
77897789
]
77907790
)

0 commit comments

Comments
 (0)