We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 85b77e5 commit 7159b28Copy full SHA for 7159b28
caffe2/CMakeLists.txt
@@ -694,7 +694,7 @@ if(USE_ROCM)
694
add_library(caffe2_nvrtc SHARED ${ATen_NVRTC_STUB_SRCS})
695
target_link_libraries(caffe2_nvrtc ${PYTORCH_HIP_LIBRARIES} ${ROCM_HIPRTC_LIB})
696
target_include_directories(caffe2_nvrtc PRIVATE ${CMAKE_BINARY_DIR})
697
- target_compile_definitions(caffe2_nvrtc PRIVATE USE_ROCM __HIP_PLATFORM_HCC__)
+ target_compile_definitions(caffe2_nvrtc PRIVATE USE_ROCM __HIP_PLATFORM_AMD__)
698
install(TARGETS caffe2_nvrtc DESTINATION "${TORCH_INSTALL_LIB_DIR}")
699
endif()
700
0 commit comments