build error error: unknown type name 'svuint8_t'

[JaneConan]

Description

INFO: Analyzed target //jaxlib/tools:build_wheel (259 packages loaded, 25906 targets configured).
INFO: Found 1 target...
ERROR: /root/.cache/bazel/_bazel_root/5ec7b40f6fc30761e523f0c22847f24b/external/compute_library/BUILD.bazel:355:11: Compiling src/core/NEON/kernels/NEBatchNormalizationLayerKernel.cpp failed: (Exit 1): clang-12 failed: error executing command (from target @compute_library//:arm_compute) 
  (cd /root/.cache/bazel/_bazel_root/5ec7b40f6fc30761e523f0c22847f24b/execroot/__main__ && \
  exec env - \
    CLANG_COMPILER_PATH=/usr/bin/clang-12 \
    LD_LIBRARY_PATH=/usr/local/Ascend/ascend-toolkit/latest/tools/aml/lib64:/usr/local/Ascend/ascend-toolkit/latest/tools/aml/lib64/plugin:/usr/local/Ascend/ascend-toolkit/latest/lib64:/usr/local/Ascend/ascend-toolkit/latest/lib64/plugin/opskernel:/usr/local/Ascend/ascend-toolkit/latest/lib64/plugin/nnengine:/usr/local/Ascend/ascend-toolkit/latest/opp/built-in/op_impl/ai_core/tbe/op_tiling/lib/linux/aarch64:/var/davinci/driver/lib64:/var/davinci/driver/lib64/common:/var/davinci/driver/lib64/driver:/usr/local/Ascend/ascend-toolkit/latest/tools/aml/lib64:/usr/local/Ascend/ascend-toolkit/latest/tools/aml/lib64/plugin:/usr/local/Ascend/ascend-toolkit/latest/lib64:/usr/local/Ascend/ascend-toolkit/latest/lib64/plugin/opskernel:/usr/local/Ascend/ascend-toolkit/latest/lib64/plugin/nnengine:/usr/local/Ascend/ascend-toolkit/latest/opp/built-in/op_impl/ai_core/tbe/op_tiling/lib/linux/aarch64:/var/davinci/driver/lib64:/var/davinci/driver/lib64/common:/var/davinci/driver/lib64/driver:/usr/local/Ascend/ascend-toolkit/latest/tools/aml/lib64:/usr/local/Ascend/ascend-toolkit/latest/tools/aml/lib64/plugin:/usr/local/Ascend/ascend-toolkit/latest/lib64:/usr/local/Ascend/ascend-toolkit/latest/lib64/plugin/opskernel:/usr/local/Ascend/ascend-toolkit/latest/lib64/plugin/nnengine:/usr/local/Ascend/ascend-toolkit/latest/opp/built-in/op_impl/ai_core/tbe/op_tiling/lib/linux/aarch64:/var/davinci/driver/lib64:/var/davinci/driver/lib64/common:/var/davinci/driver/lib64/driver:/home/HwHiAiUser/Ascend/ascend-toolkit/latest/tools/aml/lib64:/home/HwHiAiUser/Ascend/ascend-toolkit/latest/tools/aml/lib64/plugin:/home/HwHiAiUser/Ascend/ascend-toolkit/latest/lib64:/home/HwHiAiUser/Ascend/ascend-toolkit/latest/lib64/plugin/opskernel:/home/HwHiAiUser/Ascend/ascend-toolkit/latest/lib64/plugin/nnengine:/home/HwHiAiUser/Ascend/ascend-toolkit/latest/opp/built-in/op_impl/ai_core/tbe/op_tiling/lib/linux/aarch64:/var/davinci/driver/lib64:/var/davinci/driver/lib64/common:/var/davinci/driver/lib64/driver: \
    PATH=/root/miniconda3/envs/jaxenv/bin:/root/miniconda3/condabin:/root/anaconda3/bin:/usr/local/work/cmake-3.28.3-linux-aarch64/bin:/usr/local/Ascend/ascend-toolkit/latest/bin:/usr/local/Ascend/ascend-toolkit/latest/compiler/ccec_compiler/bin:/usr/local/Ascend/ascend-toolkit/latest/tools/ccec_compiler/bin:/usr/local/work/cmake-3.28.3-linux-aarch64/bin:/usr/local/Ascend/ascend-toolkit/latest/bin:/usr/local/Ascend/ascend-toolkit/latest/compiler/ccec_compiler/bin:/usr/local/Ascend/ascend-toolkit/latest/tools/ccec_compiler/bin:/usr/local/work/cmake-3.28.3-linux-aarch64/bin:/usr/local/Ascend/ascend-toolkit/latest/bin:/usr/local/Ascend/ascend-toolkit/latest/compiler/ccec_compiler/bin:/usr/local/Ascend/ascend-toolkit/latest/tools/ccec_compiler/bin:/home/HwHiAiUser/work/cmake-3.28.3-linux-aarch64/bin:/home/HwHiAiUser/Ascend/ascend-toolkit/latest/bin:/home/HwHiAiUser/Ascend/ascend-toolkit/latest/compiler/ccec_compiler/bin:/home/HwHiAiUser/Ascend/ascend-toolkit/latest/tools/ccec_compiler/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin \
    PWD=/proc/self/cwd \
  /usr/bin/clang-12 -U_FORTIFY_SOURCE -fstack-protector -Wall -Wthread-safety -Wself-assign -Wno-free-nonheap-object -fcolor-diagnostics -fno-omit-frame-pointer -g0 -O2 '-D_FORTIFY_SOURCE=1' -DNDEBUG -ffunction-sections -fdata-sections '-std=c++14' -MD -MF bazel-out/aarch64-opt/bin/external/compute_library/_objs/arm_compute/NEBatchNormalizationLayerKernel.pic.d '-frandom-seed=bazel-out/aarch64-opt/bin/external/compute_library/_objs/arm_compute/NEBatchNormalizationLayerKernel.pic.o' -fPIC -DENABLE_NEON -DARM_COMPUTE_CPU_ENABLED -DARM_COMPUTE_ENABLE_NEON -DARM_COMPUTE_ENABLE_I8MM -DENABLE_FP32_KERNELS -DENABLE_QASYMM8_KERNELS -DENABLE_QASYMM8_SIGNED_KERNELS -DENABLE_QSYMM16_KERNELS -DENABLE_INTEGER_KERNELS -DENABLE_NHWC_KERNELS -DENABLE_NCHW_KERNELS -DARM_COMPUTE_GRAPH_ENABLED -DARM_COMPUTE_ENABLE_SVEF32MM -DARM_COMPUTE_ENABLE_FIXED_FORMAT_KERNELS -D_GLIBCXX_USE_NANOSLEEP -DENABLE_SVE -DARM_COMPUTE_ENABLE_SVE -DARM_COMPUTE_ENABLE_BF16 '-DBAZEL_CURRENT_REPOSITORY="compute_library"' -iquote external/compute_library -iquote bazel-out/aarch64-opt/bin/external/compute_library -Ibazel-out/aarch64-opt/bin/external/compute_library/include/_virtual_includes/include -isystem external/compute_library/arm_compute/runtime -isystem bazel-out/aarch64-opt/bin/external/compute_library/arm_compute/runtime -isystem external/compute_library/src/core/NEON/kernels/assembly -isystem bazel-out/aarch64-opt/bin/external/compute_library/src/core/NEON/kernels/assembly -isystem external/compute_library/src/core/NEON/kernels/convolution/common -isystem bazel-out/aarch64-opt/bin/external/compute_library/src/core/NEON/kernels/convolution/common -isystem external/compute_library/src/core/NEON/kernels/convolution/winograd -isystem bazel-out/aarch64-opt/bin/external/compute_library/src/core/NEON/kernels/convolution/winograd -isystem external/compute_library/src/core/cpu/kernels/assembly -isystem bazel-out/aarch64-opt/bin/external/compute_library/src/core/cpu/kernels/assembly -isystem external/compute_library/src/cpu/kernels/assembly -isystem bazel-out/aarch64-opt/bin/external/compute_library/src/cpu/kernels/assembly -isystem external/compute_library/src/core/NEON/kernels/arm_conv -isystem bazel-out/aarch64-opt/bin/external/compute_library/src/core/NEON/kernels/arm_conv -isystem external/compute_library/src/core/NEON/kernels/arm_gemm -isystem bazel-out/aarch64-opt/bin/external/compute_library/src/core/NEON/kernels/arm_gemm '-fvisibility=hidden' -Wno-sign-compare -Wno-unknown-warning-option -Wno-stringop-truncation -Wno-array-parameter '-DMLIR_PYTHON_PACKAGE_PREFIX=jaxlib.mlir.' '-std=c++17' '-march=armv8-a' -O3 -no-canonical-prefixes -Wno-builtin-macro-redefined '-D__DATE__="redacted"' '-D__TIMESTAMP__="redacted"' '-D__TIME__="redacted"' -c external/compute_library/src/core/NEON/kernels/NEBatchNormalizationLayerKernel.cpp -o bazel-out/aarch64-opt/bin/external/compute_library/_objs/arm_compute/NEBatchNormalizationLayerKernel.pic.o)
# Configuration: aa3cae5bee02b91b5cd6c24fef447352f0e1cacb60c08e47db9131070d0f0e67
# Execution platform: @local_execution_config_platform//:platform
In file included from external/compute_library/src/core/NEON/kernels/NEBatchNormalizationLayerKernel.cpp:37:
In file included from external/compute_library/src/core/NEON/kernels/detail/NEActivationFunctionDetail.h:27:
In file included from external/compute_library/src/core/NEON/wrapper/wrapper.h:28:
In file included from external/compute_library/src/core/NEON/wrapper/traits.h:30:
/usr/lib64/clang/12.0.1/include/arm_sve.h:15:2: error: "SVE support not enabled"
#error "SVE support not enabled"
 ^
In file included from external/compute_library/src/core/NEON/kernels/NEBatchNormalizationLayerKernel.cpp:37:
In file included from external/compute_library/src/core/NEON/kernels/detail/NEActivationFunctionDetail.h:27:
In file included from external/compute_library/src/core/NEON/wrapper/wrapper.h:28:
external/compute_library/src/core/NEON/wrapper/traits.h:123:83: error: unknown type name 'svuint8_t'
template <> struct sve_vector<uint8_t>{ using scalar_type = uint8_t; using type = svuint8_t; };
                                                                                  ^
external/compute_library/src/core/NEON/wrapper/traits.h:124:81: error: unknown type name 'svint8_t'
template <> struct sve_vector<int8_t>{ using scalar_type = int8_t; using type = svint8_t; };
                                                                                ^
3 errors generated.
Target //jaxlib/tools:build_wheel failed to build
INFO: Elapsed time: 92.839s, Critical Path: 31.99s
INFO: 5 processes: 5 internal.
FAILED: Build did NOT complete successfully
ERROR: Build failed. Not running target
2025-01-09 13:47:57,341 - DEBUG - Command finished with return code 1
Traceback (most recent call last):
  File "/root/repos/jax/build/build.py", line 654, in <module>
    asyncio.run(main())
  File "/root/miniconda3/envs/jaxenv/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
  File "/root/miniconda3/envs/jaxenv/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete
    return future.result()
  File "/root/repos/jax/build/build.py", line 647, in main
    raise RuntimeError(f"Command failed with return code {result.return_code}")
RuntimeError: Command failed with return code 1
(jaxenv) sh-5.1# 

System info (python version, jaxlib version, accelerator, etc.)

Python3.10.x
ARM64

[jakevdp]

Hi - thanks for including the traceback! It looks like this error is not directly releated to JAX, but rather is coming from the ARM compute library: https://github.com/ARM-software/ComputeLibrary/blob/32bcced2af7feea6969dd1d22e58d0718dc488e3/src/core/NEON/wrapper/traits.h#L133

You might try filing an issue there, or looking for other help forums associated with those tools.

[jakevdp]

It may be that your compiler does not support SVE instructions. What compiler version are you using?

[hawkinsp]

Clang 12 is almost certainly too old to build JAX, I note. You probably want clang 17 or 18 at this point.