We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 02e0a0e commit 7cb2f8eCopy full SHA for 7cb2f8e
1 file changed
rapid_table/engine_cfg.yaml
@@ -8,7 +8,7 @@ onnxruntime:
8
9
use_cuda: false
10
cuda_ep_cfg:
11
- gpu_id: 0
+ device_id: 0
12
arena_extend_strategy: "kNextPowerOfTwo"
13
cudnn_conv_algo_search: "EXHAUSTIVE"
14
do_copy_in_default_stream: true
@@ -18,7 +18,7 @@ onnxruntime:
18
19
use_cann: false
20
cann_ep_cfg:
21
22
23
npu_mem_limit: 21474836480 # 20 * 1024 * 1024 * 1024
24
op_select_impl_mode: "high_performance"
0 commit comments