From 49412761be4fb0fb889068668097d1028c110ef8 Mon Sep 17 00:00:00 2001 From: dou <15529241576@163.com> Date: Tue, 24 Dec 2024 03:52:30 +0000 Subject: [PATCH] Fix the bug of C++ sample in CANN EP file --- .../community-maintained/CANN-ExecutionProvider.md | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/docs/execution-providers/community-maintained/CANN-ExecutionProvider.md b/docs/execution-providers/community-maintained/CANN-ExecutionProvider.md index 0f9fb14f4fff8..5802de03b5cd1 100644 --- a/docs/execution-providers/community-maintained/CANN-ExecutionProvider.md +++ b/docs/execution-providers/community-maintained/CANN-ExecutionProvider.md @@ -256,6 +256,7 @@ void output_postprocess(std::vector& output_tensors) { */ void inference() { const auto& api = Ort::GetApi(); + Ort::Env env(ORT_LOGGING_LEVEL_WARNING); // Enable cann graph in cann provider option. OrtCANNProviderOptions* cann_options = nullptr; @@ -276,7 +277,7 @@ void inference() { api.SessionOptionsAppendExecutionProvider_CANN( static_cast(session_options), cann_options); - Ort::Session session(Ort::Env(), model_path, session_options); + Ort::Session session(env, model_path, session_options); Ort::AllocatorWithDefaultOptions allocator;