提交migraphx C++代码推理

Src/Classifier.cpp

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