Speedup dumpOnArgLevel by using lazy initialization (DeepLink-org#524)

brianlcy123 · Dec 21, 2023 · 11893e8 · 11893e8
1 parent e9d098a
commit 11893e8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dipu/torch_dipu/csrc_dipu/aten/ops/OpUtils.hpp b/dipu/torch_dipu/csrc_dipu/aten/ops/OpUtils.hpp
@@ -35,8 +35,8 @@ inline void synchronizeIfEnable() {
 }
 
 inline int dumpOpArgLevel() {
-  const char* env_ptr = std::getenv("DIPU_DUMP_OP_ARGS");
-  int level = env_ptr ? std::atoi(env_ptr) : 0;
+  static const char* env_ptr = std::getenv("DIPU_DUMP_OP_ARGS");
+  static int level = env_ptr ? std::atoi(env_ptr) : 0;
   return level;
 }