Remove SPMD logging to indicate that we're using virtual device

ShouldUseVirtualDevice is called pretty liberally throughout ptxla code and it creates a lot of log spam
pytorch · Dec 18, 2024 · 8d4933d · 8d4933d
1 parent d3ed982
commit 8d4933d
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/torch_xla/csrc/device.cpp b/torch_xla/csrc/device.cpp
@@ -81,9 +81,6 @@ bool ShouldUseVirtualDevice() {
   bool use_virtual_device =
       runtime::sys_util::GetEnvBool("XLA_USE_SPMD", false) ||
       runtime::sys_util::GetEnvBool("XLA_AUTO_SPMD", false);
-  if (use_virtual_device) {
-    TF_LOG(INFO) << "Using SPMD virtual device optimization";
-  }
   return use_virtual_device;
 }