File: IntrinsicAnnotator.cpp

package info (click to toggle)
intel-graphics-compiler2 2.28.4-4
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 792,744 kB
  • sloc: cpp: 5,761,745; ansic: 466,928; lisp: 312,143; python: 114,790; asm: 44,736; pascal: 10,930; sh: 8,033; perl: 7,914; ml: 3,625; awk: 3,523; yacc: 2,747; javascript: 2,667; lex: 1,898; f90: 1,028; cs: 573; xml: 474; makefile: 344; objc: 162
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";
    }
  }
}