From 8d4933d4ce31d8ad6f0a63270f867ae552435ca1 Mon Sep 17 00:00:00 2001
From: Bhavya Bahl <bbahl@google.com>
Date: Tue, 17 Dec 2024 17:51:14 -0800
Subject: [PATCH] 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
---
 torch_xla/csrc/device.cpp | 3 ---
 1 file changed, 3 deletions(-)

diff --git a/torch_xla/csrc/device.cpp b/torch_xla/csrc/device.cpp
index a6490778477..4d1653a406d 100644
--- 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;
 }