Skip to content

Commit

Permalink
[Mosaic:TPU][NFC] Clean up unused variable
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 709171851
  • Loading branch information
tlongeri authored and Google-ML-Automation committed Dec 24, 2024
1 parent 4452960 commit ffc3c13
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions jaxlib/mosaic/dialect/tpu/transforms/apply_vector_layout.cc
Original file line number Diff line number Diff line change
Expand Up @@ -5479,8 +5479,6 @@ FailureOr<xla::Array<Value>> doColumnShiftRelayout(
const std::array<int64_t, 2> vreg_slice = src.vregSlice(target_shape);
const int bitwidth = src.bitwidth();
const int packing = src.packing();
const VectorLayout dst(bitwidth, {src.offsets()[0], dst_col_offset}, tiling,
src.implicit_dim());
const int64_t col_diff = dst_col_offset - *src.offsets()[1];
if (tiling[0] % packing != 0 || tiling[1] != target_shape[1]) {
return emitError(loc,
Expand Down

0 comments on commit ffc3c13

Please sign in to comment.