Skip to content

Commit be0c6a4

Browse files
committed
set cudnn_conv_algo_search to HEURISTIC
1 parent f0cdfd3 commit be0c6a4

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

‎plugins/Driver/onnx/internal/ExecutionProvider.cpp‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,7 @@ namespace srt::driver::onnx {
6363
constexpr int CUDA_OPTIONS_SIZE = 2;
6464
const char *cudaOptionsKeys[CUDA_OPTIONS_SIZE] = {"device_id",
6565
"cudnn_conv_algo_search"};
66-
const char *cudaOptionsValues[CUDA_OPTIONS_SIZE] = {cudaDeviceIdCStr, "DEFAULT"};
66+
const char *cudaOptionsValues[CUDA_OPTIONS_SIZE] = {cudaDeviceIdCStr, "HEURISTIC"};
6767
Ort::Status updateStatus(ortApi.UpdateCUDAProviderOptions(
6868
cudaOptions.get(), cudaOptionsKeys, cudaOptionsValues, CUDA_OPTIONS_SIZE));
6969
if (!updateStatus.IsOK()) {

0 commit comments

Comments
 (0)