Skip to content

Commit 7159b28

Browse files
committed
Removal of __HIP_PLATFORM_HCC__
1 parent 85b77e5 commit 7159b28

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

caffe2/CMakeLists.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -694,7 +694,7 @@ if(USE_ROCM)
694694
add_library(caffe2_nvrtc SHARED ${ATen_NVRTC_STUB_SRCS})
695695
target_link_libraries(caffe2_nvrtc ${PYTORCH_HIP_LIBRARIES} ${ROCM_HIPRTC_LIB})
696696
target_include_directories(caffe2_nvrtc PRIVATE ${CMAKE_BINARY_DIR})
697-
target_compile_definitions(caffe2_nvrtc PRIVATE USE_ROCM __HIP_PLATFORM_HCC__)
697+
target_compile_definitions(caffe2_nvrtc PRIVATE USE_ROCM __HIP_PLATFORM_AMD__)
698698
install(TARGETS caffe2_nvrtc DESTINATION "${TORCH_INSTALL_LIB_DIR}")
699699
endif()
700700

0 commit comments

Comments
 (0)