Build broken for AARCH64 by XLA PJRT
Broken commit was reverted by https://github.com/tensorflow/tensorflow/commit/341a0320140f94a57eb81bbba8f70615afca161e Are you satisfied with the resolution of your issue?
<a href="https://docs.google.com/forms/d/e/1FAIpQLSfaP12TRhd9xSxjXZjcZFNXPGk4kc1-qMdv3gc6bEP90vY1ew/viewform?entry.85265664=Yes&entry.2137816233=https://github.com/tensorflow/tensorflow/issues/62136">Yes</a>
<a href="https://docs.google.com/forms/d/e/1FAIpQLSfaP12TRhd9xSxjXZjcZFNXPGk4kc1-qMdv3gc6bEP90vY1ew/viewform?entry.85265664=No&entry.2137816233=https://github.com/tensorflow/tensorflow/issues/62136">No</a>

### Issue type

Bug

### Have you reproduced the bug with TensorFlow Nightly?

Yes

### Source

source

### TensorFlow version

git HEAD

### Custom code

No

### OS platform and distribution

Ubuntu 20.04

### Mobile device

n/a

### Python version

3.9.17

### Bazel version

6.1.0

### GCC/compiler version

17.0.0

### CUDA/cuDNN version

n/a

### GPU model and memory

n/a

### Current behavior?

Build fails on AARCH64 since https://github.com/tensorflow/tensorflow/commit/7b81b8c94a3fcf067ce50c31ce031a9ea449a59b

### Standalone code to reproduce the issue

```shell
bazel build --config=mkl_aarch64_threadpool --copt=-flax-vector-conversions --define=tf_api_version=2 --verbose_failures --jobs=75 -- //tensorflow/tools/pip_package:build_pip_package
```


### Relevant log output

```shell
ERROR: /home/andrew/src/tf_test/tensorflow-git/bazel-ci_build-cache/.cache/bazel/_bazel_andrew/eab0d61a99b6696edb3d2aff87b585e8/external/local_xla/xla/pjrt/BUILD:754:11: Compiling xla/pjrt/transpose.cc failed: (Exit 1): clang failed: error executing command (from target @local_xla//xla/pjrt:transpose) 
  (cd /home/andrew/src/tf_test/tensorflow-git/bazel-ci_build-cache/.cache/bazel/_bazel_andrew/eab0d61a99b6696edb3d2aff87b585e8/execroot/org_tensorflow && \
  exec env - \
    CACHEBUSTER=20220325 \
    CLANG_COMPILER_PATH=/usr/lib/llvm-17/bin/clang \
    PATH=/home/andrew/src/tf_test/tensorflow-git/bazel-ci_build-cache/.cache/bazelisk/downloads/bazelbuild/bazel-6.1.0-linux-arm64/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/games:/usr/local/games:/snap/bin \
    PWD=/proc/self/cwd \
    PYTHON_BIN_PATH=/usr/bin/python3.10 \
    PYTHON_LIB_PATH=/usr/lib/python3/dist-packages \
    TF2_BEHAVIOR=1 \
  /usr/lib/llvm-17/bin/clang -MD -MF bazel-out/aarch64-opt/bin/external/local_xla/xla/pjrt/_objs/transpose/transpose.pic.d '-frandom-seed=bazel-out/aarch64-opt/bin/external/local_xla/xla/pjrt/_objs/transpose/transpose.pic.o' -DEIGEN_MPL2_ONLY '-DEIGEN_MAX_ALIGN_BYTES=64' -DHAVE_SYS_UIO_H -DTF_USE_SNAPPY '-DBAZEL_CURRENT_REPOSITORY="local_xla"' -iquote external/local_xla -iquote bazel-out/aarch64-opt/bin/external/local_xla -iquote external/com_google_absl -iquote bazel-out/aarch64-opt/bin/external/com_google_absl -iquote external/local_tsl -iquote bazel-out/aarch64-opt/bin/external/local_tsl -iquote . -iquote bazel-out/aarch64-opt/bin -iquote external/eigen_archive -iquote bazel-out/aarch64-opt/bin/external/eigen_archive -iquote external/ml_dtypes -iquote bazel-out/aarch64-opt/bin/external/ml_dtypes -iquote external/nsync -iquote bazel-out/aarch64-opt/bin/external/nsync -iquote external/double_conversion -iquote bazel-out/aarch64-opt/bin/external/double_conversion -iquote external/com_google_protobuf -iquote bazel-out/aarch64-opt/bin/external/com_google_protobuf -iquote external/snappy -iquote bazel-out/aarch64-opt/bin/external/snappy -iquote external/com_googlesource_code_re2 -iquote bazel-out/aarch64-opt/bin/external/com_googlesource_code_re2 -Ibazel-out/aarch64-opt/bin/external/ml_dtypes/_virtual_includes/float8 -Ibazel-out/aarch64-opt/bin/external/ml_dtypes/_virtual_includes/int4 -isystem third_party/eigen3/mkl_include -isystem bazel-out/aarch64-opt/bin/third_party/eigen3/mkl_include -isystem external/eigen_archive -isystem bazel-out/aarch64-opt/bin/external/eigen_archive -isystem external/ml_dtypes -isystem bazel-out/aarch64-opt/bin/external/ml_dtypes -isystem external/ml_dtypes/ml_dtypes -isystem bazel-out/aarch64-opt/bin/external/ml_dtypes/ml_dtypes -isystem external/nsync/public -isystem bazel-out/aarch64-opt/bin/external/nsync/public -isystem external/com_google_protobuf/src -isystem bazel-out/aarch64-opt/bin/external/com_google_protobuf/src -fmerge-all-constants -Wno-builtin-macro-redefined '-D__DATE__="redacted"' '-D__TIMESTAMP__="redacted"' '-D__TIME__="redacted"' -fPIC -U_FORTIFY_SOURCE '-D_FORTIFY_SOURCE=1' -fstack-protector -Wall -Wno-invalid-partial-specialization -fno-omit-frame-pointer -no-canonical-prefixes -DNDEBUG -g0 -O2 -ffunction-sections -fdata-sections -Wno-all -Wno-extra -Wno-deprecated -Wno-deprecated-declarations -Wno-ignored-attributes -Wno-array-bounds -Wunused-result '-Werror=unused-result' -Wswitch '-Werror=switch' '-Wno-error=unused-but-set-variable' -DAUTOLOAD_DYNAMIC_KERNELS -Wno-gnu-offsetof-extensions '-mtune=generic' '-march=armv8-a' -O3 -flax-vector-conversions '-std=c++17' '--sysroot=/dt10' -c external/local_xla/xla/pjrt/transpose.cc -o bazel-out/aarch64-opt/bin/external/local_xla/xla/pjrt/_objs/transpose/transpose.pic.o)
# Configuration: 05cc03ffdfad7b565350ad2b01d7497c783d20d25ca4abeda724de457d2afb5e
# Execution platform: @local_execution_config_platform//:platform
In file included from external/local_xla/xla/pjrt/transpose.cc:94:
external/local_xla/xla/pjrt/transpose_kernels.h:710:46: error: use of undeclared identifier '__m128i'
  710 |       if constexpr (sizeof(T) * bs <= sizeof(__m128i)) {
      |                                              ^
external/local_xla/xla/pjrt/transpose_kernels.h:576:19: error: static assertion failed due to requirement 'kBytesInMatrix <= sizeof(__attribute__((neon_vector_type(2))) unsigned long) * last_transpose.size()'
  576 |     static_assert(kBytesInMatrix <= sizeof(Vec128) * last_transpose.size());
      |                   ^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
external/local_xla/xla/pjrt/transpose_kernels.h:711:66: note: in instantiation of member function 'xla::Vec128RectangularTransposeMicroKernelImpl<unsigned short, 16>::Apply' requested here
  711 |         return Vec128RectangularTransposeMicroKernelImpl<T, bs>::Apply(a, lda,
      |                                                                  ^
external/local_xla/xla/pjrt/transpose.cc:170:42: note: in instantiation of member function 'xla::TransposeMicroKernel<unsigned short, 16>::Apply' requested here
  170 |       TransposeMicroKernel<T, inner_bs>::Apply(
      |                                          ^
external/local_xla/xla/pjrt/transpose.cc:410:9: note: in instantiation of function template specialization 'xla::MacroKernel<unsigned short, 16, xla::TransposePlan::Transformation::kNone>' requested here
  410 |         MacroKernel<T, const_inner_block_elems, transformation>(
      |         ^
external/local_xla/xla/pjrt/transpose.cc:459:9: note: in instantiation of function template specialization 'xla::TransposePlan::ExecuteTyped<unsigned short, xla::TransposePlan::Transformation::kNone>' requested here
  459 |         ExecuteTyped<uint16_t, Transformation::kNone>(ac, bc, nodes);
      |         ^
external/local_xla/xla/pjrt/transpose_kernels.h:576:34: note: expression evaluates to '512 <= 256'
  576 |     static_assert(kBytesInMatrix <= sizeof(Vec128) * last_transpose.size());
      |                   ~~~~~~~~~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
external/local_xla/xla/pjrt/transpose_kernels.h:471:17: error: static assertion failed due to requirement '32UL * (0 + 1) <= sizeof(__attribute__((neon_vector_type(2))) unsigned long)'
  471 |   static_assert(bytes * (lane + 1) <= sizeof(uint64x2_t));
      |                 ^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
external/local_xla/xla/pjrt/transpose_kernels.h:551:4: note: in instantiation of function template specialization 'xla::StoreElementFromVec128<32UL, 0>' requested here
  551 |   (StoreElementFromVec128</*bytes=*/bytes, lane>(b + ldb * (i + lane), x), ...);
      |    ^
external/local_xla/xla/pjrt/transpose_kernels.h:594:7: note: in instantiation of function template specialization 'xla::StoreElementsFromVec128<32UL, 0UL>' requested here
  594 |       StoreElementsFromVec128<element_size * bs>(
      |       ^
external/local_xla/xla/pjrt/transpose_kernels.h:711:66: note: in instantiation of member function 'xla::Vec128RectangularTransposeMicroKernelImpl<unsigned short, 16>::Apply' requested here
  711 |         return Vec128RectangularTransposeMicroKernelImpl<T, bs>::Apply(a, lda,
      |                                                                  ^
external/local_xla/xla/pjrt/transpose.cc:170:42: note: in instantiation of member function 'xla::TransposeMicroKernel<unsigned short, 16>::Apply' requested here
  170 |       TransposeMicroKernel<T, inner_bs>::Apply(
      |                                          ^
external/local_xla/xla/pjrt/transpose.cc:410:9: note: in instantiation of function template specialization 'xla::MacroKernel<unsigned short, 16, xla::TransposePlan::Transformation::kNone>' requested here
  410 |         MacroKernel<T, const_inner_block_elems, transformation>(
      |         ^
external/local_xla/xla/pjrt/transpose.cc:459:9: note: in instantiation of function template specialization 'xla::TransposePlan::ExecuteTyped<unsigned short, xla::TransposePlan::Transformation::kNone>' requested here
  459 |         ExecuteTyped<uint16_t, Transformation::kNone>(ac, bc, nodes);
      |         ^
external/local_xla/xla/pjrt/transpose_kernels.h:471:36: note: expression evaluates to '32 <= 16'
  471 |   static_assert(bytes * (lane + 1) <= sizeof(uint64x2_t));
      |                 ~~~~~~~~~~~~~~~~~~~^~~~~~~~~~~~~~~~~~~~~
external/local_xla/xla/pjrt/transpose_kernels.h:498:21: error: static assertion failed due to requirement '32UL == 0'
  498 |       static_assert(bytes == 0);
      |                     ^~~~~~~~~~
external/local_xla/xla/pjrt/transpose_kernels.h:576:19: error: static assertion failed due to requirement 'kBytesInMatrix <= sizeof(__attribute__((neon_vector_type(2))) unsigned long) * last_transpose.size()'
  576 |     static_assert(kBytesInMatrix <= sizeof(Vec128) * last_transpose.size());
      |                   ^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
external/local_xla/xla/pjrt/transpose_kernels.h:711:66: note: in instantiation of member function 'xla::Vec128RectangularTransposeMicroKernelImpl<unsigned int, 8>::Apply' requested here
  711 |         return Vec128RectangularTransposeMicroKernelImpl<T, bs>::Apply(a, lda,
      |                                                                  ^
external/local_xla/xla/pjrt/transpose.cc:170:42: note: in instantiation of member function 'xla::TransposeMicroKernel<unsigned int, 8>::Apply' requested here
  170 |       TransposeMicroKernel<T, inner_bs>::Apply(
      |                                          ^
external/local_xla/xla/pjrt/transpose.cc:410:9: note: in instantiation of function template specialization 'xla::MacroKernel<unsigned int, 8, xla::TransposePlan::Transformation::kNone>' requested here
  410 |         MacroKernel<T, const_inner_block_elems, transformation>(
      |         ^
external/local_xla/xla/pjrt/transpose.cc:463:11: note: in instantiation of function template specialization 'xla::TransposePlan::ExecuteTyped<unsigned int, xla::TransposePlan::Transformation::kNone>' requested here
  463 |           ExecuteTyped<uint32_t, Transformation::kNone>(ac, bc, nodes);
      |           ^
external/local_xla/xla/pjrt/transpose_kernels.h:576:34: note: expression evaluates to '256 <= 128'
  576 |     static_assert(kBytesInMatrix <= sizeof(Vec128) * last_transpose.size());
      |                   ~~~~~~~~~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
external/local_xla/xla/pjrt/transpose_kernels.h:576:19: error: static assertion failed due to requirement 'kBytesInMatrix <= sizeof(__attribute__((neon_vector_type(2))) unsigned long) * last_transpose.size()'
  576 |     static_assert(kBytesInMatrix <= sizeof(Vec128) * last_transpose.size());
      |                   ^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
external/local_xla/xla/pjrt/transpose_kernels.h:711:66: note: in instantiation of member function 'xla::Vec128RectangularTransposeMicroKernelImpl<unsigned int, 16>::Apply' requested here
  711 |         return Vec128RectangularTransposeMicroKernelImpl<T, bs>::Apply(a, lda,
      |                                                                  ^
external/local_xla/xla/pjrt/transpose.cc:170:42: note: in instantiation of member function 'xla::TransposeMicroKernel<unsigned int, 16>::Apply' requested here
  170 |       TransposeMicroKernel<T, inner_bs>::Apply(
      |                                          ^
external/local_xla/xla/pjrt/transpose.cc:410:9: note: in instantiation of function template specialization 'xla::MacroKernel<unsigned int, 16, xla::TransposePlan::Transformation::kNone>' requested here
  410 |         MacroKernel<T, const_inner_block_elems, transformation>(
      |         ^
external/local_xla/xla/pjrt/transpose.cc:463:11: note: in instantiation of function template specialization 'xla::TransposePlan::ExecuteTyped<unsigned int, xla::TransposePlan::Transformation::kNone>' requested here
  463 |           ExecuteTyped<uint32_t, Transformation::kNone>(ac, bc, nodes);
      |           ^
external/local_xla/xla/pjrt/transpose_kernels.h:576:34: note: expression evaluates to '1024 <= 256'
  576 |     static_assert(kBytesInMatrix <= sizeof(Vec128) * last_transpose.size());
      |                   ~~~~~~~~~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
external/local_xla/xla/pjrt/transpose_kernels.h:471:17: error: static assertion failed due to requirement '64UL * (0 + 1) <= sizeof(__attribute__((neon_vector_type(2))) unsigned long)'
  471 |   static_assert(bytes * (lane + 1) <= sizeof(uint64x2_t));
      |                 ^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
external/local_xla/xla/pjrt/transpose_kernels.h:551:4: note: in instantiation of function template specialization 'xla::StoreElementFromVec128<64UL, 0>' requested here
  551 |   (StoreElementFromVec128</*bytes=*/bytes, lane>(b + ldb * (i + lane), x), ...);
      |    ^
external/local_xla/xla/pjrt/transpose_kernels.h:594:7: note: in instantiation of function template specialization 'xla::StoreElementsFromVec128<64UL, 0UL>' requested here
  594 |       StoreElementsFromVec128<element_size * bs>(
      |       ^
external/local_xla/xla/pjrt/transpose_kernels.h:711:66: note: in instantiation of member function 'xla::Vec128RectangularTransposeMicroKernelImpl<unsigned int, 16>::Apply' requested here
  711 |         return Vec128RectangularTransposeMicroKernelImpl<T, bs>::Apply(a, lda,
      |                                                                  ^
external/local_xla/xla/pjrt/transpose.cc:170:42: note: in instantiation of member function 'xla::TransposeMicroKernel<unsigned int, 16>::Apply' requested here
  170 |       TransposeMicroKernel<T, inner_bs>::Apply(
      |                                          ^
external/local_xla/xla/pjrt/transpose.cc:410:9: note: in instantiation of function template specialization 'xla::MacroKernel<unsigned int, 16, xla::TransposePlan::Transformation::kNone>' requested here
  410 |         MacroKernel<T, const_inner_block_elems, transformation>(
      |         ^
external/local_xla/xla/pjrt/transpose.cc:463:11: note: in instantiation of function template specialization 'xla::TransposePlan::ExecuteTyped<unsigned int, xla::TransposePlan::Transformation::kNone>' requested here
  463 |           ExecuteTyped<uint32_t, Transformation::kNone>(ac, bc, nodes);
      |           ^
external/local_xla/xla/pjrt/transpose_kernels.h:471:36: note: expression evaluates to '64 <= 16'
  471 |   static_assert(bytes * (lane + 1) <= sizeof(uint64x2_t));
      |                 ~~~~~~~~~~~~~~~~~~~^~~~~~~~~~~~~~~~~~~~~
external/local_xla/xla/pjrt/transpose_kernels.h:498:21: error: static assertion failed due to requirement '64UL == 0'
  498 |       static_assert(bytes == 0);
      |                     ^~~~~~~~~~
external/local_xla/xla/pjrt/transpose_kernels.h:576:19: error: static assertion failed due to requirement 'kBytesInMatrix <= sizeof(__attribute__((neon_vector_type(2))) unsigned long) * last_transpose.size()'
  576 |     static_assert(kBytesInMatrix <= sizeof(Vec128) * last_transpose.size());
      |                   ^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
external/local_xla/xla/pjrt/transpose_kernels.h:711:66: note: in instantiation of member function 'xla::Vec128RectangularTransposeMicroKernelImpl<unsigned long, 4>::Apply' requested here
  711 |         return Vec128RectangularTransposeMicroKernelImpl<T, bs>::Apply(a, lda,
      |                                                                  ^
external/local_xla/xla/pjrt/transpose.cc:170:42: note: in instantiation of member function 'xla::TransposeMicroKernel<unsigned long, 4>::Apply' requested here
  170 |       TransposeMicroKernel<T, inner_bs>::Apply(
      |                                          ^
external/local_xla/xla/pjrt/transpose.cc:410:9: note: in instantiation of function template specialization 'xla::MacroKernel<unsigned long, 4, xla::TransposePlan::Transformation::kNone>' requested here
  410 |         MacroKernel<T, const_inner_block_elems, transformation>(
      |         ^
external/local_xla/xla/pjrt/transpose.cc:470:9: note: in instantiation of function template specialization 'xla::TransposePlan::ExecuteTyped<unsigned long, xla::TransposePlan::Transformation::kNone>' requested here
  470 |         ExecuteTyped<uint64_t, Transformation::kNone>(ac, bc, nodes);
      |         ^
external/local_xla/xla/pjrt/transpose_kernels.h:576:34: note: expression evaluates to '128 <= 64'
  576 |     static_assert(kBytesInMatrix <= sizeof(Vec128) * last_transpose.size());
      |                   ~~~~~~~~~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
external/local_xla/xla/pjrt/transpose_kernels.h:576:19: error: static assertion failed due to requirement 'kBytesInMatrix <= sizeof(__attribute__((neon_vector_type(2))) unsigned long) * last_transpose.size()'
  576 |     static_assert(kBytesInMatrix <= sizeof(Vec128) * last_transpose.size());
      |                   ^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
external/local_xla/xla/pjrt/transpose_kernels.h:711:66: note: in instantiation of member function 'xla::Vec128RectangularTransposeMicroKernelImpl<unsigned long, 8>::Apply' requested here
  711 |         return Vec128RectangularTransposeMicroKernelImpl<T, bs>::Apply(a, lda,
      |                                                                  ^
external/local_xla/xla/pjrt/transpose.cc:170:42: note: in instantiation of member function 'xla::TransposeMicroKernel<unsigned long, 8>::Apply' requested here
  170 |       TransposeMicroKernel<T, inner_bs>::Apply(
      |                                          ^
external/local_xla/xla/pjrt/transpose.cc:410:9: note: in instantiation of function template specialization 'xla::MacroKernel<unsigned long, 8, xla::TransposePlan::Transformation::kNone>' requested here
  410 |         MacroKernel<T, const_inner_block_elems, transformation>(
      |         ^
external/local_xla/xla/pjrt/transpose.cc:470:9: note: in instantiation of function template specialization 'xla::TransposePlan::ExecuteTyped<unsigned long, xla::TransposePlan::Transformation::kNone>' requested here
  470 |         ExecuteTyped<uint64_t, Transformation::kNone>(ac, bc, nodes);
      |         ^
external/local_xla/xla/pjrt/transpose_kernels.h:576:34: note: expression evaluates to '512 <= 128'
  576 |     static_assert(kBytesInMatrix <= sizeof(Vec128) * last_transpose.size());
      |                   ~~~~~~~~~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
external/local_xla/xla/pjrt/transpose_kernels.h:576:19: error: static assertion failed due to requirement 'kBytesInMatrix <= sizeof(__attribute__((neon_vector_type(2))) unsigned long) * last_transpose.size()'
  576 |     static_assert(kBytesInMatrix <= sizeof(Vec128) * last_transpose.size());
      |                   ^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
external/local_xla/xla/pjrt/transpose_kernels.h:711:66: note: in instantiation of member function 'xla::Vec128RectangularTransposeMicroKernelImpl<unsigned long, 16>::Apply' requested here
  711 |         return Vec128RectangularTransposeMicroKernelImpl<T, bs>::Apply(a, lda,
      |                                                                  ^
external/local_xla/xla/pjrt/transpose.cc:170:42: note: in instantiation of member function 'xla::TransposeMicroKernel<unsigned long, 16>::Apply' requested here
  170 |       TransposeMicroKernel<T, inner_bs>::Apply(
      |                                          ^
external/local_xla/xla/pjrt/transpose.cc:410:9: note: in instantiation of function template specialization 'xla::MacroKernel<unsigned long, 16, xla::TransposePlan::Transformation::kNone>' requested here
  410 |         MacroKernel<T, const_inner_block_elems, transformation>(
      |         ^
external/local_xla/xla/pjrt/transpose.cc:470:9: note: in instantiation of function template specialization 'xla::TransposePlan::ExecuteTyped<unsigned long, xla::TransposePlan::Transformation::kNone>' requested here
  470 |         ExecuteTyped<uint64_t, Transformation::kNone>(ac, bc, nodes);
      |         ^
external/local_xla/xla/pjrt/transpose_kernels.h:576:34: note: expression evaluates to '2048 <= 256'
  576 |     static_assert(kBytesInMatrix <= sizeof(Vec128) * last_transpose.size());
      |                   ~~~~~~~~~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
external/local_xla/xla/pjrt/transpose_kernels.h:471:17: error: static assertion failed due to requirement '128UL * (0 + 1) <= sizeof(__attribute__((neon_vector_type(2))) unsigned long)'
  471 |   static_assert(bytes * (lane + 1) <= sizeof(uint64x2_t));
      |                 ^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
external/local_xla/xla/pjrt/transpose_kernels.h:551:4: note: in instantiation of function template specialization 'xla::StoreElementFromVec128<128UL, 0>' requested here
  551 |   (StoreElementFromVec128</*bytes=*/bytes, lane>(b + ldb * (i + lane), x), ...);
      |    ^
external/local_xla/xla/pjrt/transpose_kernels.h:594:7: note: in instantiation of function template specialization 'xla::StoreElementsFromVec128<128UL, 0UL>' requested here
  594 |       StoreElementsFromVec128<element_size * bs>(
      |       ^
external/local_xla/xla/pjrt/transpose_kernels.h:711:66: note: in instantiation of member function 'xla::Vec128RectangularTransposeMicroKernelImpl<unsigned long, 16>::Apply' requested here
  711 |         return Vec128RectangularTransposeMicroKernelImpl<T, bs>::Apply(a, lda,
      |                                                                  ^
external/local_xla/xla/pjrt/transpose.cc:170:42: note: in instantiation of member function 'xla::TransposeMicroKernel<unsigned long, 16>::Apply' requested here
  170 |       TransposeMicroKernel<T, inner_bs>::Apply(
      |                                          ^
external/local_xla/xla/pjrt/transpose.cc:410:9: note: in instantiation of function template specialization 'xla::MacroKernel<unsigned long, 16, xla::TransposePlan::Transformation::kNone>' requested here
  410 |         MacroKernel<T, const_inner_block_elems, transformation>(
      |         ^
external/local_xla/xla/pjrt/transpose.cc:470:9: note: in instantiation of function template specialization 'xla::TransposePlan::ExecuteTyped<unsigned long, xla::TransposePlan::Transformation::kNone>' requested here
  470 |         ExecuteTyped<uint64_t, Transformation::kNone>(ac, bc, nodes);
      |         ^
external/local_xla/xla/pjrt/transpose_kernels.h:471:36: note: expression evaluates to '128 <= 16'
  471 |   static_assert(bytes * (lane + 1) <= sizeof(uint64x2_t));
      |                 ~~~~~~~~~~~~~~~~~~~^~~~~~~~~~~~~~~~~~~~~
external/local_xla/xla/pjrt/transpose_kernels.h:498:21: error: static assertion failed due to requirement '128UL == 0'
  498 |       static_assert(bytes == 0);
      |                     ^~~~~~~~~~
external/local_xla/xla/pjrt/transpose_kernels.h:576:19: error: static assertion failed due to requirement 'kBytesInMatrix <= sizeof(__attribute__((neon_vector_type(2))) unsigned long) * last_transpose.size()'
  576 |     static_assert(kBytesInMatrix <= sizeof(Vec128) * last_transpose.size());
      |                   ^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
external/local_xla/xla/pjrt/transpose_kernels.h:711:66: note: in instantiation of member function 'xla::Vec128RectangularTransposeMicroKernelImpl<xla::uint128, 2>::Apply' requested here
  711 |         return Vec128RectangularTransposeMicroKernelImpl<T, bs>::Apply(a, lda,
      |                                                                  ^
external/local_xla/xla/pjrt/transpose.cc:170:42: note: in instantiation of member function 'xla::TransposeMicroKernel<xla::uint128, 2>::Apply' requested here
  170 |       TransposeMicroKernel<T, inner_bs>::Apply(
      |                                          ^
external/local_xla/xla/pjrt/transpose.cc:410:9: note: in instantiation of function template specialization 'xla::MacroKernel<xla::uint128, 2, xla::TransposePlan::Transformation::kNone>' requested here
  410 |         MacroKernel<T, const_inner_block_elems, transformation>(
      |         ^
external/local_xla/xla/pjrt/transpose.cc:473:9: note: in instantiation of function template specialization 'xla::TransposePlan::ExecuteTyped<xla::uint128, xla::TransposePlan::Transformation::kNone>' requested here
  473 |         ExecuteTyped<uint128, Transformation::kNone>(ac, bc, nodes);
      |         ^
external/local_xla/xla/pjrt/transpose_kernels.h:576:34: note: expression evaluates to '64 <= 32'
  576 |     static_assert(kBytesInMatrix <= sizeof(Vec128) * last_transpose.size());
      |                   ~~~~~~~~~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
external/local_xla/xla/pjrt/transpose_kernels.h:576:19: error: static assertion failed due to requirement 'kBytesInMatrix <= sizeof(__attribute__((neon_vector_type(2))) unsigned long) * last_transpose.size()'
  576 |     static_assert(kBytesInMatrix <= sizeof(Vec128) * last_transpose.size());
      |                   ^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
external/local_xla/xla/pjrt/transpose_kernels.h:711:66: note: in instantiation of member function 'xla::Vec128RectangularTransposeMicroKernelImpl<xla::uint128, 4>::Apply' requested here
  711 |         return Vec128RectangularTransposeMicroKernelImpl<T, bs>::Apply(a, lda,
      |                                                                  ^
external/local_xla/xla/pjrt/transpose.cc:170:42: note: in instantiation of member function 'xla::TransposeMicroKernel<xla::uint128, 4>::Apply' requested here
  170 |       TransposeMicroKernel<T, inner_bs>::Apply(
      |                                          ^
external/local_xla/xla/pjrt/transpose.cc:410:9: note: in instantiation of function template specialization 'xla::MacroKernel<xla::uint128, 4, xla::TransposePlan::Transformation::kNone>' requested here
  410 |         MacroKernel<T, const_inner_block_elems, transformation>(
      |         ^
external/local_xla/xla/pjrt/transpose.cc:473:9: note: in instantiation of function template specialization 'xla::TransposePlan::ExecuteTyped<xla::uint128, xla::TransposePlan::Transformation::kNone>' requested here
  473 |         ExecuteTyped<uint128, Transformation::kNone>(ac, bc, nodes);
      |         ^
external/local_xla/xla/pjrt/transpose_kernels.h:576:34: note: expression evaluates to '256 <= 64'
  576 |     static_assert(kBytesInMatrix <= sizeof(Vec128) * last_transpose.size());
      |                   ~~~~~~~~~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
external/local_xla/xla/pjrt/transpose_kernels.h:576:19: error: static assertion failed due to requirement 'kBytesInMatrix <= sizeof(__attribute__((neon_vector_type(2))) unsigned long) * last_transpose.size()'
  576 |     static_assert(kBytesInMatrix <= sizeof(Vec128) * last_transpose.size());
      |                   ^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
external/local_xla/xla/pjrt/transpose_kernels.h:711:66: note: in instantiation of member function 'xla::Vec128RectangularTransposeMicroKernelImpl<xla::uint128, 8>::Apply' requested here
  711 |         return Vec128RectangularTransposeMicroKernelImpl<T, bs>::Apply(a, lda,
      |                                                                  ^
external/local_xla/xla/pjrt/transpose.cc:170:42: note: in instantiation of member function 'xla::TransposeMicroKernel<xla::uint128, 8>::Apply' requested here
  170 |       TransposeMicroKernel<T, inner_bs>::Apply(
      |                                          ^
external/local_xla/xla/pjrt/transpose.cc:410:9: note: in instantiation of function template specialization 'xla::MacroKernel<xla::uint128, 8, xla::TransposePlan::Transformation::kNone>' requested here
  410 |         MacroKernel<T, const_inner_block_elems, transformation>(
      |         ^
external/local_xla/xla/pjrt/transpose.cc:473:9: note: in instantiation of function template specialization 'xla::TransposePlan::ExecuteTyped<xla::uint128, xla::TransposePlan::Transformation::kNone>' requested here
  473 |         ExecuteTyped<uint128, Transformation::kNone>(ac, bc, nodes);
      |         ^
external/local_xla/xla/pjrt/transpose_kernels.h:576:34: note: expression evaluates to '1024 <= 128'
  576 |     static_assert(kBytesInMatrix <= sizeof(Vec128) * last_transpose.size());
      |                   ~~~~~~~~~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
external/local_xla/xla/pjrt/transpose_kernels.h:576:19: error: static assertion failed due to requirement 'kBytesInMatrix <= sizeof(__attribute__((neon_vector_type(2))) unsigned long) * last_transpose.size()'
  576 |     static_assert(kBytesInMatrix <= sizeof(Vec128) * last_transpose.size());
      |                   ^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
external/local_xla/xla/pjrt/transpose_kernels.h:711:66: note: in instantiation of member function 'xla::Vec128RectangularTransposeMicroKernelImpl<xla::uint128, 16>::Apply' requested here
  711 |         return Vec128RectangularTransposeMicroKernelImpl<T, bs>::Apply(a, lda,
      |                                                                  ^
external/local_xla/xla/pjrt/transpose.cc:170:42: note: in instantiation of member function 'xla::TransposeMicroKernel<xla::uint128, 16>::Apply' requested here
  170 |       TransposeMicroKernel<T, inner_bs>::Apply(
      |                                          ^
external/local_xla/xla/pjrt/transpose.cc:410:9: note: in instantiation of function template specialization 'xla::MacroKernel<xla::uint128, 16, xla::TransposePlan::Transformation::kNone>' requested here
  410 |         MacroKernel<T, const_inner_block_elems, transformation>(
      |         ^
external/local_xla/xla/pjrt/transpose.cc:473:9: note: in instantiation of function template specialization 'xla::TransposePlan::ExecuteTyped<xla::uint128, xla::TransposePlan::Transformation::kNone>' requested here
  473 |         ExecuteTyped<uint128, Transformation::kNone>(ac, bc, nodes);
      |         ^
external/local_xla/xla/pjrt/transpose_kernels.h:576:34: note: expression evaluates to '4096 <= 256'
  576 |     static_assert(kBytesInMatrix <= sizeof(Vec128) * last_transpose.size());
      |                   ~~~~~~~~~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
external/local_xla/xla/pjrt/transpose_kernels.h:471:17: error: static assertion failed due to requirement '256UL * (0 + 1) <= sizeof(__attribute__((neon_vector_type(2))) unsigned long)'
  471 |   static_assert(bytes * (lane + 1) <= sizeof(uint64x2_t));
      |                 ^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
external/local_xla/xla/pjrt/transpose_kernels.h:551:4: note: in instantiation of function template specialization 'xla::StoreElementFromVec128<256UL, 0>' requested here
  551 |   (StoreElementFromVec128</*bytes=*/bytes, lane>(b + ldb * (i + lane), x), ...);
      |    ^
external/local_xla/xla/pjrt/transpose_kernels.h:594:7: note: in instantiation of function template specialization 'xla::StoreElementsFromVec128<256UL, 0UL>' requested here
  594 |       StoreElementsFromVec128<element_size * bs>(
      |       ^
external/local_xla/xla/pjrt/transpose_kernels.h:711:66: note: in instantiation of member function 'xla::Vec128RectangularTransposeMicroKernelImpl<xla::uint128, 16>::Apply' requested here
  711 |         return Vec128RectangularTransposeMicroKernelImpl<T, bs>::Apply(a, lda,
      |                                                                  ^
external/local_xla/xla/pjrt/transpose.cc:170:42: note: in instantiation of member function 'xla::TransposeMicroKernel<xla::uint128, 16>::Apply' requested here
  170 |       TransposeMicroKernel<T, inner_bs>::Apply(
      |                                          ^
external/local_xla/xla/pjrt/transpose.cc:410:9: note: in instantiation of function template specialization 'xla::MacroKernel<xla::uint128, 16, xla::TransposePlan::Transformation::kNone>' requested here
  410 |         MacroKernel<T, const_inner_block_elems, transformation>(
      |         ^
external/local_xla/xla/pjrt/transpose.cc:473:9: note: in instantiation of function template specialization 'xla::TransposePlan::ExecuteTyped<xla::uint128, xla::TransposePlan::Transformation::kNone>' requested here
  473 |         ExecuteTyped<uint128, Transformation::kNone>(ac, bc, nodes);
      |         ^
external/local_xla/xla/pjrt/transpose_kernels.h:471:36: note: expression evaluates to '256 <= 16'
  471 |   static_assert(bytes * (lane + 1) <= sizeof(uint64x2_t));
      |                 ~~~~~~~~~~~~~~~~~~~^~~~~~~~~~~~~~~~~~~~~
external/local_xla/xla/pjrt/transpose_kernels.h:498:21: error: static assertion failed due to requirement '256UL == 0'
  498 |       static_assert(bytes == 0);
      |                     ^~~~~~~~~~
19 errors generated.
Target //tensorflow/tools/pip_package:build_pip_package failed to build
```
