AI/rocm-systems
2
0
Fork 0
程式碼 問題 合併請求 Actions 套件 專案 版本發布 Wiki 動態

[HIPIFY][perl] generateSymbolFunctions() small refactoring

瀏覽代碼
此提交包含在：
Evgeny Mankov
2019-09-22 22:34:07 +03:00
父節點 9401c3a290
當前提交 4acefa95c9
共有 1 個檔案被更改，包括 16 行新增 和 24 行删除
下載 Patch 檔 下載差異檔 展開所有檔案 折疊所有檔案
hipify-clang/src/CUDA2HIP_Perl.cpp
+16 -24
查看文件
@@ -48,32 +48,24 @@ namespace perl {
void generateSymbolFunctions(std::unique_ptr<std::ostream>& perlStreamPtr) {
*perlStreamPtr.get() << "\n" << sSub << " transformSymbolFunctions\n" << "{\n" << space << sMy;
std::string sCommon = space + sForeach;
*perlStreamPtr.get() << sCommon;
unsigned int count = 0;
for (auto& dsf : DeviceSymbolFunctions0) {
const auto found = CUDA_RENAMES_MAP().find(dsf);
if (found != CUDA_RENAMES_MAP().end()) {
*perlStreamPtr.get() << (count ? ",\n" : "") << double_space << "\"" << found->second.hipName.str() << "\"";
count++;
std::set<std::string> &funcSet = DeviceSymbolFunctions0;
for (int i = 0; i < 2; ++i) {
*perlStreamPtr.get() << sCommon;
if (i == 1) funcSet = DeviceSymbolFunctions1;
unsigned int count = 0;
for (auto& f : funcSet) {
const auto found = CUDA_RUNTIME_FUNCTION_MAP.find(f);
if (found != CUDA_RUNTIME_FUNCTION_MAP.end()) {
*perlStreamPtr.get() << (count ? ",\n" : "") << double_space << "\"" << found->second.hipName.str() << "\"";
count++;
}
}
*perlStreamPtr.get() << "\n" << space << ")\n";
*perlStreamPtr.get() << space << "{\n" << double_space;
if (i ==0) *perlStreamPtr.get() << "$m += s/(?<!\\/\\/ CHECK: )($func)\\s*\\(\\s*([^,]+)\\s*,/$func\\(HIP_SYMBOL\\($2\\),/g\n";
else *perlStreamPtr.get() << "$m += s/(?<!\\/\\/ CHECK: )($func)\\s*\\(\\s*([^,]+)\\s*,\\s*([^,\\)]+)\\s*(,\\s*|\\))\\s*/$func\\($2, HIP_SYMBOL\\($3\\)$4/g;\n";
*perlStreamPtr.get() << space << "}\n";
}
*perlStreamPtr.get() << "\n" << space << ")\n";
*perlStreamPtr.get() << space << "{\n";
*perlStreamPtr.get() << double_space << "$m += s/(?<!\\/\\/ CHECK: )($func)\\s*\\(\\s*([^,]+)\\s*,/$func\\(HIP_SYMBOL\\($2\\),/g\n";
*perlStreamPtr.get() << space << "}\n";
*perlStreamPtr.get() << sCommon;
count = 0;
for (auto& dsf : DeviceSymbolFunctions1) {
const auto found = CUDA_RENAMES_MAP().find(dsf);
if (found != CUDA_RENAMES_MAP().end()) {
*perlStreamPtr.get() << (count ? ",\n" : "") << double_space << "\"" << found->second.hipName.str() << "\"";
count++;
}
}
*perlStreamPtr.get() << "\n" << space << ")\n";
*perlStreamPtr.get() << space << "{\n";
*perlStreamPtr.get() << double_space << "$m += s/(?<!\\/\\/ CHECK: )($func)\\s*\\(\\s*([^,]+)\\s*,\\s*([^,\\)]+)\\s*(,\\s*|\\))\\s*/$func\\($2, HIP_SYMBOL\\($3\\)$4/g;\n";
*perlStreamPtr.get() << space << "}\n";
*perlStreamPtr.get() << space << sReturn_m;
*perlStreamPtr.get() << "}\n";
}