[tf-frontend] print error message for TF_LoadLibrary

bytedance · Jul 10, 2024 · fb8f510 · fb8f510
1 parent 9b989ae
commit fb8f510
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/frontends/tf-frontend/tools/tf_frontend_main.cc b/frontends/tf-frontend/tools/tf_frontend_main.cc
@@ -180,6 +180,9 @@ int main(int argc, char **argv) {
     TF_Status *status_load = TF_NewStatus();
     TF_Library *lib_handle = TF_LoadLibrary(path.c_str(), status_load);
     if (!lib_handle) {
+      const char *msg = TF_Message(status_load);
+      if (nullptr != msg)
+        llvm::errs() << msg << "\n";
       llvm::errs() << "Load external library failed, path is: " << path << "\n";
       return 1;
     }