Skip to content

Commit 5119182

Browse files
authored
Merge pull request #36 from FrankWhh/fix-onnx-cuda-param
fix onnx-cuda-id-param
2 parents 2d30dc8 + ad1ce7c commit 5119182

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

rapid_layout/configs/engine_cfg.yaml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@ onnxruntime:
88

99
use_cuda: false
1010
cuda_ep_cfg:
11-
gpu_id: 0
11+
device_id: 0
1212
arena_extend_strategy: "kNextPowerOfTwo"
1313
cudnn_conv_algo_search: "EXHAUSTIVE"
1414
do_copy_in_default_stream: true
@@ -18,7 +18,7 @@ onnxruntime:
1818

1919
use_cann: false
2020
cann_ep_cfg:
21-
gpu_id: 0
21+
device_id: 0
2222
arena_extend_strategy: "kNextPowerOfTwo"
2323
npu_mem_limit: 21474836480 # 20 * 1024 * 1024 * 1024
2424
op_select_impl_mode: "high_performance"

0 commit comments

Comments
 (0)