From 9411550a9a1100e94b0f391e0e23821fcd3149fd Mon Sep 17 00:00:00 2001 From: Vincent Wang Date: Thu, 14 Apr 2022 10:31:23 +0800 Subject: [PATCH] Fix Key Number in CUDA EP Sample Code (#11203) * fix key number in doc * use size() instead of number --- docs/execution-providers/CUDA-ExecutionProvider.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/execution-providers/CUDA-ExecutionProvider.md b/docs/execution-providers/CUDA-ExecutionProvider.md index 2d6a003ed4..3a7b0912dd 100644 --- a/docs/execution-providers/CUDA-ExecutionProvider.md +++ b/docs/execution-providers/CUDA-ExecutionProvider.md @@ -149,7 +149,7 @@ CreateCUDAProviderOptions(&cuda_options); std::vector keys{"device_id", "gpu_mem_limit", "arena_extend_strategy", "cudnn_conv_algo_search", "do_copy_in_default_stream", "cudnn_conv_use_max_workspace", "cudnn_conv1d_pad_to_nc1d"}; std::vector values{"0", "2147483648", "kSameAsRequested", "DEFAULT", "1", "1", "1"}; -UpdateCUDAProviderOptions(cuda_options, keys.data(), values.data(), 6); +UpdateCUDAProviderOptions(cuda_options, keys.data(), values.data(), keys.size()); OrtSessionOptions* session_options = /* ... */; SessionOptionsAppendExecutionProvider_CUDA_V2(session_options, cuda_options);