Skip to content

Commit b24d870

Browse files
committed
add fixes for python and java
1 parent 75b51e4 commit b24d870

File tree

3 files changed

+52
-2
lines changed

3 files changed

+52
-2
lines changed

.github/workflows/windows_webgpu.yml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -121,6 +121,7 @@ jobs:
121121
--build_shared_lib `
122122
--enable_onnx_tests `
123123
--build_nodejs `
124+
--build_java `
124125
--use_webgpu `
125126
${{ matrix.vcpkg_option == 'vcpkg' && '--use_vcpkg' || '' }} `
126127
--cmake_extra_defines `

cmake/onnxruntime_java.cmake

Lines changed: 21 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -175,8 +175,27 @@ if (WIN32)
175175
if (onnxruntime_USE_QNN AND NOT onnxruntime_BUILD_QNN_EP_STATIC_LIB)
176176
add_custom_command(TARGET onnxruntime4j_jni POST_BUILD COMMAND ${CMAKE_COMMAND} -E copy_if_different $<TARGET_FILE:onnxruntime_providers_qnn> ${JAVA_PACKAGE_LIB_DIR}/$<TARGET_FILE_NAME:onnxruntime_providers_qnn>)
177177
endif()
178-
if (onnxruntime_USE_WEBGPU AND onnxruntime_BUILD_DAWN_MONOLITHIC_LIBRARY)
179-
add_custom_command(TARGET onnxruntime4j_jni POST_BUILD COMMAND ${CMAKE_COMMAND} -E copy_if_different $<TARGET_FILE:dawn::webgpu_dawn> ${JAVA_PACKAGE_LIB_DIR}/$<TARGET_FILE_NAME:dawn::webgpu_dawn>)
178+
if (onnxruntime_USE_WEBGPU)
179+
if (onnxruntime_ENABLE_DAWN_BACKEND_D3D12)
180+
if (onnxruntime_USE_VCPKG)
181+
add_custom_command(
182+
TARGET onnxruntime4j_jni POST_BUILD COMMAND ${CMAKE_COMMAND} -E copy_if_different
183+
$<TARGET_FILE:Microsoft::DXIL>
184+
$<TARGET_FILE:Microsoft::DirectXShaderCompiler>
185+
${JAVA_PACKAGE_LIB_DIR}/
186+
)
187+
else()
188+
add_custom_command(
189+
TARGET onnxruntime4j_jni POST_BUILD COMMAND ${CMAKE_COMMAND} -E copy_if_different
190+
$<TARGET_FILE_DIR:dxcompiler>/dxil.dll
191+
$<TARGET_FILE_DIR:dxcompiler>/dxcompiler.dll
192+
${JAVA_PACKAGE_LIB_DIR}/
193+
)
194+
endif()
195+
endif()
196+
if (onnxruntime_BUILD_DAWN_MONOLITHIC_LIBRARY)
197+
add_custom_command(TARGET onnxruntime4j_jni POST_BUILD COMMAND ${CMAKE_COMMAND} -E copy_if_different $<TARGET_FILE:dawn::webgpu_dawn> ${JAVA_PACKAGE_LIB_DIR}/$<TARGET_FILE_NAME:dawn::webgpu_dawn>)
198+
endif()
180199
endif()
181200
endif()
182201
else()

cmake/onnxruntime_python.cmake

Lines changed: 30 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1073,6 +1073,36 @@ if (onnxruntime_USE_QNN)
10731073
endif()
10741074
endif()
10751075

1076+
if (onnxruntime_USE_WEBGPU)
1077+
if (WIN32 AND onnxruntime_ENABLE_DAWN_BACKEND_D3D12)
1078+
if (onnxruntime_USE_VCPKG)
1079+
add_custom_command(
1080+
TARGET onnxruntime_pybind11_state POST_BUILD
1081+
COMMAND ${CMAKE_COMMAND} -E copy
1082+
$<TARGET_FILE:Microsoft::DXIL>
1083+
$<TARGET_FILE:Microsoft::DirectXShaderCompiler>
1084+
$<TARGET_FILE_DIR:${build_output_target}>/onnxruntime/capi/
1085+
)
1086+
else()
1087+
add_custom_command(
1088+
TARGET onnxruntime_pybind11_state POST_BUILD
1089+
COMMAND ${CMAKE_COMMAND} -E copy
1090+
$<TARGET_FILE_DIR:dxcompiler>/dxil.dll
1091+
$<TARGET_FILE_DIR:dxcompiler>/dxcompiler.dll
1092+
$<TARGET_FILE_DIR:${build_output_target}>/onnxruntime/capi/
1093+
)
1094+
endif()
1095+
endif()
1096+
if (onnxruntime_BUILD_DAWN_MONOLITHIC_LIBRARY)
1097+
add_custom_command(
1098+
TARGET onnxruntime_pybind11_state POST_BUILD
1099+
COMMAND ${CMAKE_COMMAND} -E copy
1100+
$<TARGET_FILE:dawn::webgpu_dawn>
1101+
$<TARGET_FILE_DIR:${build_output_target}>/onnxruntime/capi/
1102+
)
1103+
endif()
1104+
endif()
1105+
10761106
if (onnxruntime_USE_VSINPU)
10771107
add_custom_command(
10781108
TARGET onnxruntime_pybind11_state POST_BUILD

0 commit comments

Comments
 (0)