SWDEV-341360 - Add Catch2 loweredname hipRTC test (#2801)

Change-Id: Ibe8736951af491e48dee77644bbd0475d326c152
2022-07-19 19:27:14 +05:30
parent 0fc518d281
commit 9be55a6379
1 changed files with 33 additions and 0 deletions
@@ -146,3 +146,36 @@ TEST_CASE("Unit_hiprtc_namehandling") {
  hiprtcDestroyProgram(&prog);
  REQUIRE(compileResult == HIPRTC_SUCCESS);
 }
+
+TEST_CASE("Unit_hiprtc_getloweredname") {
+  using namespace std;
+  hiprtcProgram prog;
+  hiprtcCreateProgram(&prog,                 // prog
+                      template_kernel,       // buffer
+                      "template_kernel.cu",  // name
+                      0, nullptr, nullptr);
+
+  std::string name_expression = "my_sqrt<complex<double> >";
+  REQUIRE(HIPRTC_SUCCESS == hiprtcAddNameExpression(prog, name_expression.c_str()));
+
+  hiprtcResult compileResult{hiprtcCompileProgram(prog, 0, 0)};
+
+  size_t logSize;
+  HIPRTC_CHECK(hiprtcGetProgramLogSize(prog, &logSize));
+  if (logSize) {
+    string log(logSize, '\0');
+    HIPRTC_CHECK(hiprtcGetProgramLog(prog, &log[0]));
+    std::cout << log << '\n';
+  }
+
+  const char* mangled_instantiation_cstr;
+  // Verifies if hiprtcGetLoweredName successfully gets the lowered name for named expressions with space
+  REQUIRE(HIPRTC_SUCCESS == hiprtcGetLoweredName(prog, name_expression.c_str(), &mangled_instantiation_cstr));
+
+  std::string mangled_name_str = mangled_instantiation_cstr;
+  // Checks if the fetched lowered name is not empty
+  REQUIRE(mangled_name_str.size() > 0);
+
+  hiprtcDestroyProgram(&prog);
+  REQUIRE(compileResult == HIPRTC_SUCCESS);
+}