From c67816d65bea16fe1acf79e9ac7505452eae65f1 Mon Sep 17 00:00:00 2001 From: Xinpeng Dou <15529241576@163.com> Date: Sun, 29 Dec 2024 14:13:47 +0800 Subject: [PATCH] Fix the bug of C++ sample in CANN EP file (#23186) ### Description 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 0f9fb14f4f..5802de03b5 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;