Output tensorid before erroring out for empty data handle (#7176)

pytorch · Jun 4, 2024 · 376d645 · 376d645
1 parent bcf46c7
commit 376d645
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/torch_xla/csrc/tensor.cpp b/torch_xla/csrc/tensor.cpp
@@ -214,8 +214,8 @@ torch::lazy::BackendDataPtr XLATensor::GetXlaData() {
     torch::lazy::BackendDataPtr handle = CurrentDataHandle();
     if (handle != nullptr) {
       XLA_CHECK(handle->HasValue())
-          << "Trying to access XLA data while an async operation is in flight: "
-          << handle->shape();
+          << "Trying to access XLA data for tensor with ID " << GetUniqueId()
+          << " while an async operation is in flight: " << handle->shape();
       return handle;
     }
   }