File: IntrinsicAnnotator.cpp

package info (click to toggle)
intel-graphics-compiler2 2.16.0-2
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 106,644 kB
  • sloc: cpp: 805,640; lisp: 287,672; ansic: 16,414; python: 3,952; yacc: 2,588; lex: 1,666; pascal: 313; sh: 186; makefile: 35
file content (45 lines) | stat: -rw-r--r-- 1,431 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
/*========================== begin_copyright_notice ============================

Copyright (C) 2020-2022 Intel Corporation

SPDX-License-Identifier: MIT

============================= end_copyright_notice ===========================*/

#include "IntrinsicAnnotator.hpp"
#include "common/LLVMWarningsPush.hpp"
#include <llvm/Support/FormattedStream.h>
#include "common/LLVMWarningsPop.hpp"
#include "GenISAIntrinsics/GenIntrinsics.h"

#include <sstream>

using namespace llvm;
using namespace IGC;
using namespace GenISAIntrinsic;

void IntrinsicAnnotator::emitFunctionAnnot(const Function *func, formatted_raw_ostream &OS) {
  if (isIntrinsic(func)) {
    auto AddCommentPrefixes = [](const char *in) {
      std::stringstream ss(in);
      std::string content;
      if (std::getline(ss, content, '\n')) {
        std::string line;
        while (std::getline(ss, line, '\n')) {
          content += "\n; " + line;
        }
      }
      return content;
    };

    auto ID = getIntrinsicID(func);
    IntrinsicComments comments = getIntrinsicComments(ID);
    OS << "; Function Desc: " << AddCommentPrefixes(comments.funcDescription) << "\n";
    for (auto out : comments.outputs) {
      OS << "; Output: " << AddCommentPrefixes(out) << "\n";
    }
    for (std::vector<int>::size_type i = 0; i != comments.inputs.size(); i++) {
      OS << "; Arg " << i << ": " << AddCommentPrefixes(comments.inputs[i]) << "\n";
    }
  }
}