mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-21 13:29:20 +00:00
213
host/online_compile/addkernels/include_inliner.cpp
Normal file
213
host/online_compile/addkernels/include_inliner.cpp
Normal file
@@ -0,0 +1,213 @@
|
||||
/*******************************************************************************
|
||||
*
|
||||
* MIT License
|
||||
*
|
||||
* Copyright (c) 2021 Advanced Micro Devices, Inc.
|
||||
*
|
||||
* Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||
* of this software and associated documentation files (the "Software"), to deal
|
||||
* in the Software without restriction, including without limitation the rights
|
||||
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||
* copies of the Software, and to permit persons to whom the Software is
|
||||
* furnished to do so, subject to the following conditions:
|
||||
*
|
||||
* The above copyright notice and this permission notice shall be included in all
|
||||
* copies or substantial portions of the Software.
|
||||
*
|
||||
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
||||
* SOFTWARE.
|
||||
*
|
||||
*******************************************************************************/
|
||||
#include <algorithm>
|
||||
#include <exception>
|
||||
#include <fstream>
|
||||
#include <sstream>
|
||||
|
||||
#ifdef _WIN32
|
||||
#include <windows.h>
|
||||
#endif
|
||||
#ifdef __linux__
|
||||
#include <linux/limits.h>
|
||||
#include <cstdlib>
|
||||
#endif // !WIN32
|
||||
|
||||
#include "include_inliner.hpp"
|
||||
|
||||
namespace PathHelpers {
|
||||
static int GetMaxPath()
|
||||
{
|
||||
#ifdef _WIN32
|
||||
return MAX_PATH;
|
||||
#else
|
||||
return PATH_MAX;
|
||||
#endif
|
||||
}
|
||||
|
||||
static std::string GetAbsolutePath(const std::string& path)
|
||||
{
|
||||
std::string result(GetMaxPath(), ' ');
|
||||
#ifdef _WIN32
|
||||
const auto retval = GetFullPathName(path.c_str(), result.size(), &result[0], nullptr);
|
||||
|
||||
if(retval == 0)
|
||||
return "";
|
||||
#else
|
||||
auto* const retval = realpath(path.c_str(), &result[0]);
|
||||
|
||||
if(retval == nullptr)
|
||||
return "";
|
||||
#endif
|
||||
return result;
|
||||
}
|
||||
} // namespace PathHelpers
|
||||
|
||||
std::string IncludeFileExceptionBase::What() const
|
||||
{
|
||||
std::ostringstream ss;
|
||||
ss << GetMessage() << ": <" << _file << ">";
|
||||
|
||||
return ss.str();
|
||||
}
|
||||
|
||||
void IncludeInliner::Process(std::istream& input,
|
||||
std::ostream& output,
|
||||
const std::string& root,
|
||||
const std::string& file_name,
|
||||
const std::string& directive,
|
||||
bool allow_angle_brackets,
|
||||
bool recurse)
|
||||
{
|
||||
ProcessCore(input, output, root, file_name, 0, directive, allow_angle_brackets, recurse);
|
||||
}
|
||||
|
||||
void IncludeInliner::ProcessCore(std::istream& input,
|
||||
std::ostream& output,
|
||||
const std::string& root,
|
||||
const std::string& file_name,
|
||||
int line_number,
|
||||
const std::string& directive,
|
||||
bool allow_angle_brackets,
|
||||
bool recurse)
|
||||
{
|
||||
if(_include_depth >= include_depth_limit)
|
||||
throw InlineStackOverflowException(GetIncludeStackTrace(0));
|
||||
|
||||
_include_depth++;
|
||||
_included_stack_head =
|
||||
std::make_shared<SourceFileDesc>(file_name, _included_stack_head, line_number);
|
||||
auto current_line = 0;
|
||||
auto next_include_optional = false;
|
||||
|
||||
while(!input.eof())
|
||||
{
|
||||
std::string line;
|
||||
std::string word;
|
||||
std::getline(input, line);
|
||||
std::istringstream line_parser(line);
|
||||
line_parser >> word;
|
||||
current_line++;
|
||||
std::transform(word.begin(), word.end(), word.begin(), ::tolower);
|
||||
|
||||
const auto include_optional = next_include_optional;
|
||||
next_include_optional = false;
|
||||
|
||||
if(!word.empty() && word == "//inliner-include-optional")
|
||||
{
|
||||
if(include_optional)
|
||||
throw IncludeExpectedException(GetIncludeStackTrace(current_line));
|
||||
next_include_optional = true;
|
||||
continue;
|
||||
}
|
||||
|
||||
if(!word.empty() && word == directive && recurse)
|
||||
{
|
||||
auto first_quote_pos = line.find('"', static_cast<int>(line_parser.tellg()) + 1);
|
||||
std::string::size_type second_quote_pos;
|
||||
|
||||
if(first_quote_pos != std::string::npos)
|
||||
{
|
||||
second_quote_pos = line.find('"', first_quote_pos + 1);
|
||||
if(second_quote_pos == std::string::npos)
|
||||
throw WrongInlineDirectiveException(GetIncludeStackTrace(current_line));
|
||||
}
|
||||
else
|
||||
{
|
||||
if(!allow_angle_brackets)
|
||||
throw WrongInlineDirectiveException(GetIncludeStackTrace(current_line));
|
||||
|
||||
first_quote_pos = line.find('<', static_cast<int>(line_parser.tellg()) + 1);
|
||||
if(first_quote_pos == std::string::npos)
|
||||
throw WrongInlineDirectiveException(GetIncludeStackTrace(current_line));
|
||||
|
||||
second_quote_pos = line.find('>', first_quote_pos + 1);
|
||||
if(second_quote_pos == std::string::npos)
|
||||
throw WrongInlineDirectiveException(GetIncludeStackTrace(current_line));
|
||||
}
|
||||
|
||||
const std::string include_file_path =
|
||||
line.substr(first_quote_pos + 1, second_quote_pos - first_quote_pos - 1);
|
||||
const std::string abs_include_file_path(
|
||||
PathHelpers::GetAbsolutePath(root + "/" + include_file_path)); // NOLINT
|
||||
|
||||
if(abs_include_file_path.empty())
|
||||
{
|
||||
if(include_optional)
|
||||
continue;
|
||||
throw IncludeNotFoundException(include_file_path,
|
||||
GetIncludeStackTrace(current_line));
|
||||
}
|
||||
std::ifstream include_file(abs_include_file_path, std::ios::in);
|
||||
|
||||
if(!include_file.good())
|
||||
throw IncludeCantBeOpenedException(include_file_path,
|
||||
GetIncludeStackTrace(current_line));
|
||||
|
||||
ProcessCore(include_file,
|
||||
output,
|
||||
root,
|
||||
include_file_path,
|
||||
current_line,
|
||||
directive,
|
||||
allow_angle_brackets,
|
||||
recurse);
|
||||
}
|
||||
else
|
||||
{
|
||||
if(include_optional)
|
||||
throw IncludeExpectedException(GetIncludeStackTrace(current_line));
|
||||
|
||||
if(output.tellp() > 0)
|
||||
output << std::endl;
|
||||
|
||||
output << line;
|
||||
}
|
||||
}
|
||||
|
||||
auto prev_file = _included_stack_head->included_from;
|
||||
_included_stack_head = prev_file;
|
||||
_include_depth--;
|
||||
}
|
||||
|
||||
std::string IncludeInliner::GetIncludeStackTrace(int line)
|
||||
{
|
||||
std::ostringstream ss;
|
||||
|
||||
if(_included_stack_head == nullptr)
|
||||
return "";
|
||||
|
||||
auto item = _included_stack_head;
|
||||
ss << " " << item->path << ":" << line;
|
||||
|
||||
while(item->included_from != nullptr)
|
||||
{
|
||||
ss << std::endl << " from " << item->included_from->path << ":" << item->included_line;
|
||||
item = item->included_from;
|
||||
}
|
||||
|
||||
return ss.str();
|
||||
}
|
||||
Reference in New Issue
Block a user