mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-07-14 18:12:05 +00:00
12 lines
473 B
C++
12 lines
473 B
C++
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
// Licensed under the MIT License.
|
|
|
|
#include "utils.h"
|
|
|
|
OrtCUDAProviderOptions CreateDefaultOrtCudaProviderOptionsWithCustomStream(void* cuda_compute_stream) {
|
|
OrtCUDAProviderOptions cuda_options;
|
|
cuda_options.do_copy_in_default_stream = true;
|
|
cuda_options.has_user_compute_stream = cuda_compute_stream != nullptr ? 1 : 0;
|
|
cuda_options.user_compute_stream = cuda_compute_stream;
|
|
return cuda_options;
|
|
}
|