Safely assign xtensor to prevent build error

hshahTT · hshahTT · commit 6f3cf07b14f3 · 2025-10-03T17:04:57.000Z
diff --git a/torch_xla/csrc/init_python_bindings.cpp b/torch_xla/csrc/init_python_bindings.cpp
@@ -761,7 +761,7 @@ std::string GetTensorsHloGraph(const std::vector<at::Tensor>& tensors,
 }
 
 std::optional<xla::OpSharding> GetXLAOpSharding(const at::Tensor& input) {
-  XLATensorPtr xtensor = bridge::GetXlaTensor(input);
+  XLA_ASSIGN_OR_THROW(XLATensorPtr xtensor, bridge::GetXlaTensor(input));
   XLATensor::ShardingSpecPtr sharding_spec =
       xtensor ? xtensor->sharding_spec() : nullptr;
   if (sharding_spec != nullptr) {

Original file line number	Diff line number	Diff line change
`@@ -761,7 +761,7 @@ std::string GetTensorsHloGraph(const std::vector<at::Tensor>& tensors,`
`761`	`761`	`}`
`762`	`762`
`763`	`763`	`std::optional<xla::OpSharding> GetXLAOpSharding(const at::Tensor& input) {`
`764`		`- XLATensorPtr xtensor = bridge::GetXlaTensor(input);`
	`764`	`+ XLA_ASSIGN_OR_THROW(XLATensorPtr xtensor, bridge::GetXlaTensor(input));`
`765`	`765`	`XLATensor::ShardingSpecPtr sharding_spec =`
`766`	`766`	`xtensor ? xtensor->sharding_spec() : nullptr;`
`767`	`767`	`if (sharding_spec != nullptr) {`