Skip to content

Commit

Permalink
Automated Code Change
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 660572431
  • Loading branch information
tensorflower-gardener authored and copybara-github committed Aug 7, 2024
1 parent 3bf7e1a commit 56d83b8
Show file tree
Hide file tree
Showing 2 changed files with 1 addition and 3 deletions.
1 change: 0 additions & 1 deletion xla/hlo/utils/hlo_sharding_util.cc
Original file line number Diff line number Diff line change
Expand Up @@ -2247,7 +2247,6 @@ std::optional<GatherScatterParallelDims> GetGatherScatterBatchParallelDims(
// %indices = concatenate(..., %iota.1, ...)
// ... = gather(..., %indices)
// is common for tf.reverse_sequence and would match this case.
absl::InlinedVector<const HloIotaInstruction*, 4> iotas;
const int num_indices = index_map.size();
std::vector<int64_t> index_parallel_in_dim(num_indices, -1);

Expand Down
3 changes: 1 addition & 2 deletions xla/service/algebraic_simplifier.cc
Original file line number Diff line number Diff line change
Expand Up @@ -3341,12 +3341,11 @@ AlgebraicSimplifierVisitor::AssociativeReorderDotOperator(
} else if (opcode == HloOpcode::kBroadcast) {
// Broadcasts of dot contracting dimensions can be reordered to reduces
// of the corresponding contracting dimensions in the other dot operand
DimensionVector reduce_dims, broadcast_dim_sizes;
DimensionVector reduce_dims;
const int64_t pre_broadcast_rank =
reorder_from->mutable_operand(0)->shape().rank();
int64_t post_broadcast_rank = reorder_from->shape().rank();
Shape new_broadcast_shape = reorder_from->shape();
DimensionVector contracting_reordered;

// Construct map from broadcasted shape to its original shape. Broadcast
// dimensions are mapped to -1 since they were not present
Expand Down

0 comments on commit 56d83b8

Please sign in to comment.