Skip to content

Commit 5e4af41

Browse files
committed
Debug CI
1 parent 56651ec commit 5e4af41

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

src/backends/onnxruntime.c

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -329,7 +329,7 @@ RAI_Model *RAI_ModelCreateORT(RAI_Backend backend, const char *devicestr, RAI_Mo
329329
}
330330

331331
// TODO: these options could be configured at the AI.CONFIG level
332-
// ONNX_VALIDATE_STATUS(ort->SetSessionGraphOptimizationLevel(session_options, 1))
332+
ONNX_VALIDATE_STATUS(ort->SetSessionGraphOptimizationLevel(session_options, ORT_DISABLE_ALL))
333333
ONNX_VALIDATE_STATUS(
334334
ort->SetIntraOpNumThreads(session_options, (int)opts.backends_intra_op_parallelism))
335335
ONNX_VALIDATE_STATUS(

tests/flow/tests_onnx.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -504,7 +504,7 @@ def test_onnx_use_custom_allocator(env):
504504
ai_memory_config = {k.split(":")[0]: k.split(":")[1]
505505
for k in con.execute_command("INFO MODULES").decode().split("#")[4].split()[1:]}
506506
allocator_access_num_after = ai_memory_config["ai_onnxruntime_memory_access_num"]
507-
env.assertEqual(int(allocator_access_num_after) - int(allocator_access_num_before), 16)
507+
env.assertEqual(int(allocator_access_num_after) - int(allocator_access_num_before), 22)
508508

509509
values = con.execute_command('AI.TENSORGET', 'b{1}', 'VALUES')
510510
argmax = max(range(len(values)), key=lambda i: values[i])

0 commit comments

Comments
 (0)