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.
2 parents 2d30dc8 + ad1ce7c commit 5119182Copy full SHA for 5119182
rapid_layout/configs/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