Skip to content

Commit f3f4d78

Browse files
committed
Fix lint issues.
1 parent 91e1a72 commit f3f4d78

File tree

3 files changed

+10
-9
lines changed

3 files changed

+10
-9
lines changed

test/cpp/cpp_test_util.cpp

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -295,9 +295,11 @@ std::vector<torch_xla::runtime::ComputationClient::DataPtr> Execute(
295295
std::move(instances));
296296

297297
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));
301303
}
302304

303305
std::vector<at::Tensor> Fetch(

test/cpp/test_replication.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -65,13 +65,13 @@ void TestSingleReplication(
6565
torch_xla::runtime::ComputationClient::ExecuteComputationOptions exec_options;
6666
for (size_t i = 0; i < device_strings.size(); ++i) {
6767
auto executor = [&, i]() {
68-
results[i] =
68+
results[i] = GetValueOrThrow(
6969
torch_xla::runtime::GetComputationClientOrDie()->ExecuteComputation(
7070
*compiled_computations[i],
7171
{std::dynamic_pointer_cast<
7272
torch_xla::runtime::ComputationClient::Data>(
7373
tensors_data[i])},
74-
device_strings[i], exec_options);
74+
device_strings[i], exec_options));
7575
counter.DecrementCount();
7676
};
7777
torch_xla::thread::Schedule(std::move(executor));

torch_xla/csrc/runtime/pjrt_computation_client.cpp

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -874,10 +874,9 @@ PjRtComputationClient::ExecuteReplicated(
874874
tsl::profiler::TraceMe activity(
875875
"PjRtComputationClient::ExecuteReplicated_execute",
876876
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));
881880

882881
(*returned_futures)[0].OnReady(
883882
std::move([timed, op_tracker = std::move(op_tracker)](

0 commit comments

Comments
 (0)