Update onnxruntime.cmake

PaddlePaddle · Nov 21, 2023 · 12803a7 · 12803a7
1 parent cfebd24
commit 12803a7
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/cmake/onnxruntime.cmake b/cmake/onnxruntime.cmake
@@ -48,7 +48,9 @@ if(WIN32)
   if(WITH_GPU)
     set(ONNXRUNTIME_FILENAME "onnxruntime-win-x64-gpu-${ONNXRUNTIME_VERSION}.zip")
   elseif(WITH_DIRECTML)
-    set(ONNXRUNTIME_FILENAME "onnxruntime-directml-win-x64.zip")
+    # set(ONNXRUNTIME_FILENAME "onnxruntime-directml-win-x86.zip")
+    set(ONNXRUNTIME_FILENAME "onnxruntime-directml-win-x86-1.16.2.zip")
+    set(ONNXRUNTIME_VERSION "1.16.2")
   else()
     set(ONNXRUNTIME_FILENAME "onnxruntime-win-x64-${ONNXRUNTIME_VERSION}.zip")
   endif()
@@ -126,4 +128,4 @@ endif()
 
 add_library(external_onnxruntime STATIC IMPORTED GLOBAL)
 set_property(TARGET external_onnxruntime PROPERTY IMPORTED_LOCATION ${ONNXRUNTIME_LIB})
-add_dependencies(external_onnxruntime ${ONNXRUNTIME_PROJECT})
+add_dependencies(external_onnxruntime ${ONNXRUNTIME_PROJECT})