• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 /**
2  * Copyright 2020 Huawei Technologies Co., Ltd
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  * http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
17 #ifndef MINDSPORE_CCSRC_RUNTIME_DEVICE_ASCEND_PROFILING_REPORTER_DESC_REPORTER_H_
18 #define MINDSPORE_CCSRC_RUNTIME_DEVICE_ASCEND_PROFILING_REPORTER_DESC_REPORTER_H_
19 
20 #include <string>
21 #include <utility>
22 #include <vector>
23 #include <memory>
24 #include "toolchain/prof_reporter.h"
25 #include "runtime/device/ascend/profiling/reporter/profiling_desc.h"
26 #include "utils/contract.h"
27 #include "backend/session/kernel_graph.h"
28 
29 namespace mindspore {
30 namespace device {
31 namespace ascend {
32 class DescReporter {
33  public:
34   virtual ~DescReporter() = 0;
DescReporter(int device_id,std::string file_name)35   DescReporter(int device_id, std::string file_name) : device_id_(device_id), file_name_(std::move(file_name)) {}
36 
37   virtual void ReportData() = 0;
38 
39  protected:
40   void ReportByLine(const std::string &data, const std::string &file_name) const;
41   void ReportAllLine();
42 
43   int device_id_;
44   std::string file_name_;
45   std::vector<std::shared_ptr<ProfDesc>> prof_desc_list_;
46 };
47 }  // namespace ascend
48 }  // namespace device
49 }  // namespace mindspore
50 #endif  // MINDSPORE_CCSRC_RUNTIME_DEVICE_ASCEND_PROFILING_REPORTER_DESC_REPORTER_H_
51