zzf/ disable fallback op info display (#748)

* disable fallback op info display * Update RegisterDIPU.hpp
DeepLink-org · Mar 26, 2024 · c2d8a3d · c2d8a3d
1 parent df93c05
commit c2d8a3d
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/dipu/torch_dipu/csrc_dipu/aten/RegisterDIPU.hpp b/dipu/torch_dipu/csrc_dipu/aten/RegisterDIPU.hpp
@@ -34,13 +34,13 @@ void dipu_fallback(const c10::OperatorHandle& op, DispatchKeySet dispatch_keys,
 // NOLINTEND(bugprone-macro-parentheses)
 
 // Check the environment variable and call the DIPU_LOG_WARNING_ONCE
-#define DIPU_OP_LOG_WARNING_ONCE(...)                      \
-  do {                                                     \
-    const char* env = std::getenv("DIPU_DUMP_OP_ARGS");    \
-    int env_value = (env != nullptr) ? std::atoi(env) : 0; \
-    if (env_value >= 0) {                                  \
-      DIPU_LOG_WARNING_ONCE(__VA_ARGS__);                  \
-    }                                                      \
+#define DIPU_OP_LOG_WARNING_ONCE(...)                       \
+  do {                                                      \
+    const char* env = std::getenv("DIPU_DUMP_OP_ARGS");     \
+    int env_value = (env != nullptr) ? std::atoi(env) : -1; \
+    if (env_value >= 0) {                                   \
+      DIPU_LOG_WARNING_ONCE(__VA_ARGS__);                   \
+    }                                                       \
   } while (0)
 
 // Temporarily not implement 'sub-dispatch from box' (from torch box func ->