saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-29 20:14:01 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix the bug of C++ sample in CANN EP file (#23186)

Browse source 
### Description
Fix the bug of C++ sample in CANN EP file
This commit is contained in:
Xinpeng Dou 2024-12-29 14:13:47 +08:00 committed by GitHub
parent af2f2d517c
commit c67816d65b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
docs/execution-providers/community-maintained/CANN-ExecutionProvider.md
View file

@ -256,6 +256,7 @@ void output_postprocess(std::vector<Ort::Value>& 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<OrtSessionOptions*>(session_options), cann_options);
Ort::Session session(Ort::Env(), model_path, session_options);
Ort::Session session(env, model_path, session_options);
Ort::AllocatorWithDefaultOptions allocator;