Skip to content

Commit 01c9522

Browse files
committed
check in
1 parent 76f2334 commit 01c9522

File tree

3 files changed

+14
-2
lines changed

3 files changed

+14
-2
lines changed

libkineto/include/Config.h

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -502,4 +502,8 @@ class Config : public AbstractConfig {
502502

503503
constexpr char kUseDaemonEnvVar[] = "KINETO_USE_DAEMON";
504504

505+
#if __linux__
506+
extern int kUseDaemonEnvVarSet;
507+
#endif
508+
505509
} // namespace libkineto

libkineto/src/Config.cpp

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@
88

99
#include "Config.h"
1010

11+
#include <assert.h>
1112
#include <stdlib.h>
1213

1314
#include <fmt/chrono.h>
@@ -243,7 +244,8 @@ Config::Config()
243244
factories->addFeatureConfigs(*this);
244245
}
245246
#if __linux__
246-
enableIpcFabric_ = (getenv(kUseDaemonEnvVar) != nullptr);
247+
assert(libkineto::kUseDaemonEnvVarSet >= 0);
248+
enableIpcFabric_ = libkineto::kUseDaemonEnvVarSet;
247249
#endif
248250
}
249251

libkineto/src/init.cpp

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,10 @@
3131
#include "Logger.h"
3232

3333
namespace KINETO_NAMESPACE {
34+
#if __linux__
35+
int kUseDaemonEnvVarSet = -1;
36+
#endif
37+
3438

3539
#if __linux__ || defined(HAS_CUPTI)
3640
static bool initialized = false;
@@ -131,7 +135,9 @@ void libkineto_init(bool cpuOnly, bool logOnError) {
131135

132136
// Factory to connect to open source daemon if present
133137
#if __linux__
134-
if (getenv(kUseDaemonEnvVar) != nullptr) {
138+
void *ptr = getenv(kUseDaemonEnvVar);
139+
libkineto::kUseDaemonEnvVarSet = ptr != nullptr;
140+
if (libkineto::kUseDaemonEnvVarSet) {
135141
LOG(INFO) << "Registering daemon config loader, cpuOnly = "
136142
<< cpuOnly;
137143
DaemonConfigLoader::registerFactory();

0 commit comments

Comments
 (0)