Skip to content

Commit 1713204

Browse files
committed
GPU CUDA ORT: Fix usage of OrtCUDAProviderOptionsV2
1 parent e9b2d16 commit 1713204

File tree

3 files changed

+3
-4
lines changed

3 files changed

+3
-4
lines changed

GPU/GPUTracking/Base/cuda/CMakeLists.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -114,7 +114,6 @@ if(ALIGPU_BUILD_TYPE STREQUAL "O2")
114114
${MODULE}
115115
SOURCES ${SRCS}
116116
PUBLIC_LINK_LIBRARIES O2::GPUTracking O2::ITStrackingCUDA
117-
PRIVATE_LINK_LIBRARIES onnxruntime::onnxruntime
118117
PRIVATE_INCLUDE_DIRECTORIES
119118
${CMAKE_SOURCE_DIR}/Detectors/Base/src
120119
${CMAKE_SOURCE_DIR}/Detectors/TRD/base/src
@@ -139,6 +138,7 @@ if (onnxruntime_FOUND)
139138
target_compile_definitions(${targetName} PRIVATE
140139
$<$<BOOL:${ORT_CUDA_BUILD}>:ORT_CUDA_BUILD>
141140
$<$<BOOL:${ORT_TENSORRT_BUILD}>:ORT_TENSORRT_BUILD>)
141+
target_link_libraries(${targetName} PRIVATE onnxruntime::onnxruntime)
142142
endif()
143143

144144
# Setting target architecture and adding GPU libraries

GPU/GPUTracking/Base/cuda/GPUReconstructionCUDA.cu

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -643,9 +643,8 @@ void GPUReconstructionCUDA::SetONNXGPUStream(Ort::SessionOptions& session_option
643643
// UpdateCUDAProviderOptions(cuda_options, keys.data(), values.data(), keys.size());
644644

645645
// this implicitly sets "has_user_compute_stream"
646-
cuda_options->has_user_compute_stream = 1;
647646
ORTCHK(api->UpdateCUDAProviderOptionsWithValue(cuda_options, "user_compute_stream", mInternals->Streams[stream]));
648-
session_options.AppendExecutionProvider_CUDA_V2(cuda_options);
647+
ORTCHK(api->SessionOptionsAppendExecutionProvider_CUDA_V2(session_options, cuda_options));
649648

650649
// Finally, don't forget to release the provider options
651650
api->ReleaseCUDAProviderOptions(cuda_options);

GPU/GPUTracking/Base/hip/CMakeLists.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -162,7 +162,6 @@ if(ALIGPU_BUILD_TYPE STREQUAL "O2")
162162
${MODULE}
163163
SOURCES ${SRCS}
164164
PUBLIC_LINK_LIBRARIES O2::GPUTracking O2::ITStrackingHIP
165-
PRIVATE_LINK_LIBRARIES onnxruntime::onnxruntime
166165
PRIVATE_INCLUDE_DIRECTORIES
167166
${CMAKE_SOURCE_DIR}/Detectors/Base/src
168167
${CMAKE_SOURCE_DIR}/Detectors/TRD/base/src
@@ -194,6 +193,7 @@ if (onnxruntime_FOUND)
194193
target_compile_definitions(${targetName} PRIVATE
195194
$<$<BOOL:${ORT_ROCM_BUILD}>:ORT_ROCM_BUILD>
196195
$<$<BOOL:${ORT_MIGRAPHX_BUILD}>:ORT_MIGRAPHX_BUILD>)
196+
target_link_libraries(${targetName} PRIVATE onnxruntime::onnxruntime)
197197
endif()
198198

199199
add_library(${MODULE}_CXX OBJECT ${SRCS_CXX}) # Adding a C++ library for the .cxx code of the HIP library, such that it does not link to HIP libraries, and CMake HIP Language doesn't add HIP compile flags.

0 commit comments

Comments
 (0)