Fix enable_training and use_migraphx

microsoft · Oct 7, 2023 · 4f8a4dd · 4f8a4dd
1 parent 37f4f27
commit 4f8a4dd
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/cmake/onnxruntime_providers.cmake b/cmake/onnxruntime_providers.cmake
@@ -1434,6 +1434,14 @@ if (onnxruntime_USE_MIGRAPHX)
       message(STATUS "MIGRAPHX GPU STREAM SYNC is DISABLED")
   endif()
 
+  if (onnxruntime_ENABLE_TRAINING_OPS)
+    onnxruntime_add_include_to_target(onnxruntime_providers_migraphx onnxruntime_training)
+    target_link_libraries(onnxruntime_providers_migraphx PRIVATE onnxruntime_training)
+    if (onnxruntime_ENABLE_TRAINING_TORCH_INTEROP)
+      onnxruntime_add_include_to_target(onnxruntime_providers_migraphx Python::Module)
+    endif()
+  endif()
+
   install(TARGETS onnxruntime_providers_migraphx
           ARCHIVE  DESTINATION ${CMAKE_INSTALL_LIBDIR}
           LIBRARY  DESTINATION ${CMAKE_INSTALL_LIBDIR}