修改offload为false的示例

Src/Classifier.cpp

@@ -21,6 +21,16 @@ Classifier::~Classifier()
     configurationFile.release();
 }
 
+std::unordered_map<std::string, migraphx::argument> Classifier::CreateOutputData(migraphx::program &p)
+{
+    std::unordered_map<std::string, migraphx::argument> outputData;
+    for (auto x : p.get_outputs())
+    {
+        outputData[x.first] = migraphx::gpu::allocate_gpu(x.second);
+    } 
+    return outputData;
+}
+
 ErrorCode Classifier::Initialize(InitializationParameterOfClassifier initializationParameterOfClassifier)
 {
     // 读取配置文件

Src/Classifier.h

@@ -15,6 +15,8 @@ public:
     
     ~Classifier();
 
+    std::unordered_map<std::string, migraphx::argument> CreateOutputData(migraphx::program &p);
+
     ErrorCode Initialize(InitializationParameterOfClassifier initializationParameterOfClassifier);
 
     ErrorCode Classify(const std::vector<cv::Mat> &srcImages,std::vector<std::vector<ResultOfPrediction>> &predictions);
@@ -29,6 +31,7 @@ private:
 
     bool useInt8;
     bool useFP16;
+    bool useoffloadcopy;
     
 
 };