message: Remove the threshold on tile_size_x for column reduction from IsReductionFromOrToContiguousDimensions, so that it returns true regardless tile_size_x.
diff: @@ -220,9 +220,9 @@ bool IsReductionFromOrToContiguousDimensions(const HloInstruction& reduce) {
  }
  // For column reduction, the tile block is tize_size_y x tile_size_x, and we
  - // are reducing along tile_size_y. Both tile_size_x and tile_size_y need to be
  + // are reducing along tile_size_y. tile_size_y needs to be
  // large enough to make the tiling implementation efficient.
  - return dims_in_elem[2] >= kWarpSize && dims_in_elem[1] >= kWarpSize;
  + return dims_in_elem[1] >= kWarpSize;
  }
  std::pair<bool, DimensionVector> GetReductionKindAndContiguousComponents(
  