From c4f8772d5c7719a5d428992224233607fba8653f Mon Sep 17 00:00:00 2001
From: jonb377 <jonbolin@google.com>
Date: Wed, 13 Dec 2023 14:02:44 -0800
Subject: [PATCH] Initialize compilation env hash (#6140)

---
 torch_xla/csrc/runtime/pjrt_computation_client.cc | 1 +
 1 file changed, 1 insertion(+)

diff --git a/torch_xla/csrc/runtime/pjrt_computation_client.cc b/torch_xla/csrc/runtime/pjrt_computation_client.cc
index 187c2f35c7e1..14914b6b182f 100644
--- a/torch_xla/csrc/runtime/pjrt_computation_client.cc
+++ b/torch_xla/csrc/runtime/pjrt_computation_client.cc
@@ -244,6 +244,7 @@ PjRtComputationClient::PjRtComputationClient() {
     std::string device_str = PjRtDeviceToString(device);
     string_to_device_.emplace(device_str, device);
   }
+  comp_env_hash_ = hash_comp_env(client_, ordered_devices);
 
   auto tracked_devices = GetLocalDevices();
   tracked_devices.emplace_back(spmd_device_str);