Add more debug log when Shardy dumping is disabled.

PiperOrigin-RevId: 702267000
openxla · Dec 3, 2024 · 3c7dd7a · 3c7dd7a
1 parent a00416c
commit 3c7dd7a
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/xla/service/spmd/shardy/shardy_xla_pass.cc b/xla/service/spmd/shardy/shardy_xla_pass.cc
@@ -316,6 +316,9 @@ absl::StatusOr<bool> ShardyXLA::Run(
     if (shardyDir.empty()) {
       LOG(WARNING) << "\"sponge\" specified as dump directory but "
                       "TEST_UNDECLARED_OUTPUTS_DIR is not set!";
+    } else {
+      LOG(INFO) << "Shardy dump directory is sponge on undeclared outputs dir: "
+                << shardyDir;
     }
   }