saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-06-06 00:03:22 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix enable_training and use_migraphx (#17827)

Browse source
This commit is contained in:
cloudhan 2023-10-08 11:43:27 +08:00 committed by GitHub
parent faf9a0f6c7
commit c2bd5b70b2
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
cmake/onnxruntime_providers.cmake
View file

@ -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}