diff --git a/cmake/onnxruntime_session.cmake b/cmake/onnxruntime_session.cmake index cdd853465d..2d0b3dcb27 100644 --- a/cmake/onnxruntime_session.cmake +++ b/cmake/onnxruntime_session.cmake @@ -66,3 +66,7 @@ if (NOT onnxruntime_BUILD_SHARED_LIB) RUNTIME DESTINATION ${CMAKE_INSTALL_BINDIR} FRAMEWORK DESTINATION ${CMAKE_INSTALL_BINDIR}) endif() + +if (onnxruntime_USE_NCCL AND onnxruntime_USE_ROCM) + add_dependencies(onnxruntime_session generate_hipified_files) +endif()