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";
}
}
}
|