File tree Expand file tree Collapse file tree 3 files changed +10
-9
lines changed Expand file tree Collapse file tree 3 files changed +10
-9
lines changed Original file line number Diff line number Diff line change @@ -295,9 +295,11 @@ std::vector<torch_xla::runtime::ComputationClient::DataPtr> Execute(
295
295
std::move (instances));
296
296
297
297
torch_xla::runtime::ComputationClient::ExecuteComputationOptions options;
298
- return torch_xla::runtime::GetComputationClientOrDie ()->ExecuteComputation (
299
- *computations.front (), UnwrapXlaData (lowering_ctx.GetParametersData ()),
300
- device.toString (), options);
298
+ return GetValueOrThrow (
299
+ torch_xla::runtime::GetComputationClientOrDie ()->ExecuteComputation (
300
+ *computations.front (),
301
+ UnwrapXlaData (lowering_ctx.GetParametersData ()), device.toString (),
302
+ options));
301
303
}
302
304
303
305
std::vector<at::Tensor> Fetch (
Original file line number Diff line number Diff line change @@ -65,13 +65,13 @@ void TestSingleReplication(
65
65
torch_xla::runtime::ComputationClient::ExecuteComputationOptions exec_options;
66
66
for (size_t i = 0 ; i < device_strings.size (); ++i) {
67
67
auto executor = [&, i]() {
68
- results[i] =
68
+ results[i] = GetValueOrThrow (
69
69
torch_xla::runtime::GetComputationClientOrDie ()->ExecuteComputation (
70
70
*compiled_computations[i],
71
71
{std::dynamic_pointer_cast<
72
72
torch_xla::runtime::ComputationClient::Data>(
73
73
tensors_data[i])},
74
- device_strings[i], exec_options);
74
+ device_strings[i], exec_options)) ;
75
75
counter.DecrementCount ();
76
76
};
77
77
torch_xla::thread::Schedule (std::move (executor));
Original file line number Diff line number Diff line change @@ -874,10 +874,9 @@ PjRtComputationClient::ExecuteReplicated(
874
874
tsl::profiler::TraceMe activity (
875
875
" PjRtComputationClient::ExecuteReplicated_execute" ,
876
876
tsl::profiler::TraceMeLevel::kInfo );
877
- XLA_ASSIGN_OR_RETURN (
878
- results,
879
- pjrt_computation.executable ->Execute (
880
- std::move (argument_handles), execute_options, returned_futures));
877
+ XLA_ASSIGN_OR_RETURN (results, pjrt_computation.executable ->Execute (
878
+ std::move (argument_handles),
879
+ execute_options, returned_futures));
881
880
882
881
(*returned_futures)[0 ].OnReady (
883
882
std::move ([timed, op_tracker = std::move (op_tracker)](
You can’t perform that action at this time.
0 commit comments