提交migraphx C++代码推理

e7035bdf · liucong · 79cf04ca · e7035bdf
Commit e7035bdf authored Jan 07, 2025 by liucong
Show whitespace changes
Inline Side-by-side

Showing with 5 additions and 6 deletions

Src/Classifier.cpp Src/Classifier.cpp +5 -6

No files found.
--- a/Src/Classifier.cpp
+++ b/Src/Classifier.cpp
@@ -37,12 +37,11 @@ ErrorCode Classifier::Initialize(InitializationParameterOfClassifier initializat
    std::string modelPath=(std::string)netNode["ModelPath"];
    // 初始化session
-    //设置DCU
+    OrtMIGraphXProviderOptions migraphx_options;
-    OrtROCMProviderOptions rocm_options;
+    migraphx_options.device_id = 0;
-    rocm_options.device_id = 0;
+    migraphx_options.migraphx_fp16_enable = 1;
-    sessionOptions.AppendExecutionProvider_ROCM(rocm_options);
+    migraphx_options.dynamic_model = 0; 
-    sessionOptions.SetGraphOptimizationLevel(GraphOptimizationLevel::ORT_ENABLE_BASIC);
+    sessionOptions.AppendExecutionProvider_MIGraphX(migraphx_options);
-    // sessionOptions.EnableProfiling("profile_prefix");
    session = new Ort::Session(env, modelPath.c_str(), sessionOptions);
    return SUCCESS;