Skip to content

Commit 7d18543

Browse files
authored
fix: fixed issue #126
fix: fixed issue #126
2 parents 02e0a0e + 7cb2f8e commit 7d18543

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

rapid_table/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)