check in

pytorch · Aug 31, 2024 · 01c9522 · 01c9522
1 parent 76f2334
commit 01c9522
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 2 deletions.
diff --git a/libkineto/include/Config.h b/libkineto/include/Config.h
@@ -502,4 +502,8 @@ class Config : public AbstractConfig {
 
 constexpr char kUseDaemonEnvVar[] = "KINETO_USE_DAEMON";
 
+#if __linux__
+extern int kUseDaemonEnvVarSet;
+#endif
+
 } // namespace libkineto
diff --git a/libkineto/src/Config.cpp b/libkineto/src/Config.cpp
@@ -8,6 +8,7 @@
 
 #include "Config.h"
 
+#include <assert.h>
 #include <stdlib.h>
 
 #include <fmt/chrono.h>
@@ -243,7 +244,8 @@ Config::Config()
     factories->addFeatureConfigs(*this);
   }
 #if __linux__
-  enableIpcFabric_ = (getenv(kUseDaemonEnvVar) != nullptr);
+  assert(libkineto::kUseDaemonEnvVarSet >= 0);
+  enableIpcFabric_ = libkineto::kUseDaemonEnvVarSet;
 #endif
 }
 

diff --git a/libkineto/src/init.cpp b/libkineto/src/init.cpp
@@ -31,6 +31,10 @@
 #include "Logger.h"
 
 namespace KINETO_NAMESPACE {
+#if __linux__
+int kUseDaemonEnvVarSet = -1;
+#endif
+
 
 #if __linux__ || defined(HAS_CUPTI)
 static bool initialized = false;
@@ -131,7 +135,9 @@ void libkineto_init(bool cpuOnly, bool logOnError) {
 
   // Factory to connect to open source daemon if present
 #if __linux__
-  if (getenv(kUseDaemonEnvVar) != nullptr) {
+  void *ptr = getenv(kUseDaemonEnvVar);
+  libkineto::kUseDaemonEnvVarSet = ptr != nullptr;
+  if (libkineto::kUseDaemonEnvVarSet) {
     LOG(INFO) << "Registering daemon config loader, cpuOnly =  "
               << cpuOnly;
     DaemonConfigLoader::registerFactory();