Skip to content

Commit

Permalink
clean up
Browse files Browse the repository at this point in the history
  • Loading branch information
JackCaoG committed Aug 26, 2022
1 parent 2bd6bb2 commit bf9b61e
Showing 1 changed file with 1 addition and 7 deletions.
8 changes: 1 addition & 7 deletions torch_xla/csrc/helpers.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -609,13 +609,7 @@ xla::StatusOr<xla::XlaComputation> XlaHelpers::WrapXlaComputation(
xla::XlaOp orig_result;
orig_result = xla::Call(&builder, computation, inner_params);

// Construct a single tuple result.
const std::vector<xla::XlaOp> results = [&orig_result]() {
std::vector<xla::XlaOp> results;
results.push_back(orig_result);
return results;
}();

// Rebuild aliasing.
for (const auto& [input_index, output_index] : input_output_alias_pair) {
// Both input and output will be a tuple so parameter_number will always be
// 0
Expand Down

0 comments on commit bf9b61e

Please sign in to comment.