Safely assign xtensor to prevent build error

hshahTT · hshahTT · commit 9c0e2c06424f · 2025-09-07T06:08:19.000Z
diff --git a/torch_xla/csrc/init_python_bindings.cpp b/torch_xla/csrc/init_python_bindings.cpp
@@ -753,7 +753,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
`@@ -753,7 +753,7 @@ std::string GetTensorsHloGraph(const std::vector<at::Tensor>& tensors,`
`753`	`753`	`}`
`754`	`754`
`755`	`755`	`std::optional<xla::OpSharding> GetXLAOpSharding(const at::Tensor& input) {`
`756`		`- XLATensorPtr xtensor = bridge::GetXlaTensor(input);`
	`756`	`+ XLA_ASSIGN_OR_THROW(XLATensorPtr xtensor, bridge::GetXlaTensor(input));`
`757`	`757`	`XLATensor::ShardingSpecPtr sharding_spec =`
`758`	`758`	`xtensor ? xtensor->sharding_spec() : nullptr;`
`759`	`759`	`if (sharding_spec != nullptr) {`