From e5f7e04e6ffb70541b589a0ffccd86b9f18ea271 Mon Sep 17 00:00:00 2001 From: Tiago Koji Castro Shibata Date: Tue, 11 Feb 2020 12:54:38 -0800 Subject: [PATCH] Use target_add_dml function (#3005) * Use target_add_dml function * Don't access DML if disabled --- cmake/winml.cmake | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/cmake/winml.cmake b/cmake/winml.cmake index 4b245b7db7..6277ac4a16 100644 --- a/cmake/winml.cmake +++ b/cmake/winml.cmake @@ -180,7 +180,9 @@ add_dependencies(winml_lib_ort winml_api_native) add_dependencies(winml_lib_ort winml_api_native_internal) # Link libraries -target_link_libraries(winml_lib_ort PRIVATE ${CMAKE_CURRENT_BINARY_DIR}/packages/DirectML.0.0.1/build/DirectML.targets) +if (onnxruntime_USE_DML) + target_add_dml(winml_lib_ort) +endif() target_link_libraries(winml_lib_ort PRIVATE wil)