1 parent f0cdfd3 commit be0c6a4Copy full SHA for be0c6a4
1 file changed
plugins/Driver/onnx/internal/ExecutionProvider.cpp
@@ -63,7 +63,7 @@ namespace srt::driver::onnx {
63
constexpr int CUDA_OPTIONS_SIZE = 2;
64
const char *cudaOptionsKeys[CUDA_OPTIONS_SIZE] = {"device_id",
65
"cudnn_conv_algo_search"};
66
- const char *cudaOptionsValues[CUDA_OPTIONS_SIZE] = {cudaDeviceIdCStr, "DEFAULT"};
+ const char *cudaOptionsValues[CUDA_OPTIONS_SIZE] = {cudaDeviceIdCStr, "HEURISTIC"};
67
Ort::Status updateStatus(ortApi.UpdateCUDAProviderOptions(
68
cudaOptions.get(), cudaOptionsKeys, cudaOptionsValues, CUDA_OPTIONS_SIZE));
69
if (!updateStatus.IsOK()) {
0 commit comments