Merge pull request #1085 from emankov/master

[HIPIFY][perl] Initial support of hipify-perl generation from hipify-clang

[ROCm/clr commit: df8909d73d]

projects/clr/hipamd/hipify-clang/src/ArgParse.cpp

@@ -34,6 +34,21 @@ cl::opt<std::string> OutputDir("o-dir",
   cl::value_desc("directory"),
   cl::cat(ToolTemplateCategory));
 
+cl::opt <bool> GeneratePerl("perl",
+  cl::desc("Generate hipify-perl"),
+  cl::value_desc("perl"),
+  cl::cat(ToolTemplateCategory));
+
+cl::opt<std::string> OutputPerlMapFilename("o-perl-map",
+  cl::desc("Output filename for Perl map"),
+  cl::value_desc("filename"),
+  cl::cat(ToolTemplateCategory));
+
+cl::opt<std::string> OutputPerlMapDir("o-perl-map-dir",
+  cl::desc("Output direcory for Perl map"),
+  cl::value_desc("directory"),
+  cl::cat(ToolTemplateCategory));
+
 cl::opt<std::string> TemporaryDir("temp-dir",
   cl::desc("Temporary direcory"),
   cl::value_desc("directory"),

projects/clr/hipamd/hipify-clang/src/ArgParse.h

@@ -30,6 +30,8 @@ namespace ct = clang::tooling;
 
 extern cl::OptionCategory ToolTemplateCategory;
 extern cl::opt<std::string> OutputFilename;
+extern cl::opt<std::string> OutputPerlMapFilename;
+extern cl::opt<std::string> OutputPerlMapDir;
 extern cl::opt<std::string> OutputDir;
 extern cl::opt<std::string> TemporaryDir;
 extern cl::opt<std::string> CudaPath;
@@ -37,6 +39,7 @@ extern cl::list<std::string> IncludeDirs;
 extern cl::list<std::string> MacroNames;
 extern cl::opt<bool> Inplace;
 extern cl::opt<bool> SaveTemps;
+extern cl::opt<bool> GeneratePerl;
 extern cl::opt<bool> Verbose;
 extern cl::opt<bool> NoBackup;
 extern cl::opt<bool> NoOutput;

projects/clr/hipamd/hipify-clang/src/main.cpp

@@ -93,6 +93,57 @@ std::string getAbsoluteDirectoryPath(const std::string& sDir, std::error_code& E
   return dirAbsPath.c_str();
 }
 
+bool generatePerl(bool Generate = true) {
+  if (!Generate) {
+    return true;
+  }
+  std::string dstPerlMap = OutputPerlMapFilename, dstPerlMapDir = OutputPerlMapDir;
+  if (dstPerlMap.empty()) {
+    dstPerlMap = "hipify-perl-map";
+  }
+  std::error_code EC;
+  if (!dstPerlMapDir.empty()) {
+    std::string sOutputPerlMapDirAbsPath = getAbsoluteDirectoryPath(OutputPerlMapDir, EC, "output hipify-perl map");
+    if (EC) {
+      return false;
+    }
+    dstPerlMap = sOutputPerlMapDirAbsPath + "/" + dstPerlMap;
+  }
+  SmallString<128> tmpFile;
+  StringRef ext = "hipify-tmp";
+  EC = sys::fs::createTemporaryFile(dstPerlMap, ext, tmpFile);
+  if (EC) {
+    llvm::errs() << "\n" << sHipify << sError << EC.message() << ": " << tmpFile << "\n";
+    return false;
+  }
+  std::unique_ptr<std::ostream> perlStreamPtr = std::unique_ptr<std::ostream>(new std::ofstream(tmpFile.c_str(), std::ios_base::trunc));
+  std::string sConv = "my $conversions = ";
+  *perlStreamPtr.get() << "@statNames = (";
+  for (int i = 0; i < NUM_CONV_TYPES - 1; i++) {
+    *perlStreamPtr.get() << "\"" << counterNames[i] << "\", ";
+    sConv += "$ft{'" + std::string(counterNames[i]) + "'} + ";
+  }
+  *perlStreamPtr.get() << "\"" << counterNames[NUM_CONV_TYPES - 1] << "\");\n\n";
+  *perlStreamPtr.get() << sConv << "$ft{'" << counterNames[NUM_CONV_TYPES - 1] << "'};\n\n";
+  for (auto& ma : CUDA_RENAMES_MAP()) {
+    if (Statistics::isUnsupported(ma.second)) {
+      continue;
+    }
+    *perlStreamPtr.get() << "$ft{'" + std::string(counterNames[ma.second.type]) + "'} += s/\\b" + std::string(ma.first) + "\\b/" + std::string(ma.second.hipName) + "/g;\n";
+  }
+  perlStreamPtr.get()->flush();
+  bool ret = true;
+  EC = sys::fs::copy_file(tmpFile, dstPerlMap);
+  if (EC) {
+    llvm::errs() << "\n" << sHipify << sError << EC.message() << ": while copying " << tmpFile << " to " << dstPerlMap << "\n";
+    ret = false;
+  }
+  if (!SaveTemps) {
+    sys::fs::remove(tmpFile);
+  }
+  return ret;
+}
+
 int main(int argc, const char **argv) {
   std::vector<const char*> new_argv(argv, argv + argc);
   if (std::find(new_argv.begin(), new_argv.end(), std::string("--")) == new_argv.end()) {
@@ -102,8 +153,19 @@ int main(int argc, const char **argv) {
     argc++;
   }
   llcompat::PrintStackTraceOnErrorSignal();
-  ct::CommonOptionsParser OptionsParser(argc, argv, ToolTemplateCategory, llvm::cl::OneOrMore);
+  ct::CommonOptionsParser OptionsParser(argc, argv, ToolTemplateCategory, llvm::cl::Optional);
   std::vector<std::string> fileSources = OptionsParser.getSourcePathList();
+  if (fileSources.empty() && !GeneratePerl) {
+    llvm::errs() << "\n" << sHipify << sError << "Must specify at least 1 positional argument for source file." << "\n";
+    return 1;
+  }
+  if (!generatePerl(GeneratePerl)) {
+    llvm::errs() << "\n" << sHipify << sError << "hipify-perl generating failed." << "\n";
+    return 1;
+  }
+  if (fileSources.empty()) {
+    return 0;
+  }
   std::string dst = OutputFilename, dstDir = OutputDir;
   std::error_code EC;
   std::string sOutputDirAbsPath = getAbsoluteDirectoryPath(OutputDir, EC, "output");