buildx failed with: ERROR: failed to solve: process "/bin/sh -c <<\"EOF\" bash -ex\n# Use XLA's Bazel configuration to get the relevant tag from the openxla/triton\n# fork's llvm-head branch and apply XLA's extra patches to it. Also fetches the\n# compatible LLVM sources.\npushd \"${SRC_PATH_XLA}\"\nBAZEL=$(find \"${SRC_PATH_JAX}/build\" -type f -executable -name 'bazel-*')\n\"${BAZEL}\" --output_base=/opt/checkout fetch @triton//:BUILD\n# Build XLA's version of LLVM\nmkdir /opt/llvm-build\npushd /opt/llvm-build\ncmake -G Ninja \\\n -DCMAKE_BUILD_TYPE=Release \\\n -DCMAKE_C_COMPILER=clang \\\n -DCMAKE_CXX_COMPILER=clang++ \\\n -DLLVM_ENABLE_ASSERTIONS=ON \\\n -DLLVM_ENABLE_PROJECTS=\"mlir;llvm\" \\\n -DLLVM_TARGETS_TO_BUILD=\"host;NVPTX\" \\\n /opt/checkout/external/llvm-raw/llvm\nninja\n# Build XLA's version of Triton against that LLVM\npushd /opt/checkout/external/triton\nmkdir dist\n# Do not compile with -Werror\nsed -i -e 's|-Werror||g' CMakeLists.txt\n# The LLVM build above does not enable these libraries\nsed -i -e 's|\\(LLVMAMDGPU.*\\)|# \\1|g' CMakeLists.txt\n# Do not build tests\nsed -i -e 's|^add_subdirectory(unittest)|# unit tests disabled|' CMakeLists.txt\n# Do not build the AMD GPU backend\nsed -i -e 's|BackendInstaller.copy(\\[\"nvidia\", \"amd\"\\])|BackendInstaller.copy([\"nvidia\"])|g' python/setup.py\n# Extra patches to Triton maintained in XLA. These are already applied in the working directory.\nXLA_TRITON_PATCHES=\"${SRC_PATH_XLA}/third_party/triton\"\n# This patch adds two files that are not known to CMake\nif [[ -f \"${XLA_TRITON_PATCHES}/xla_extensions/sparse_dot_base.patch\" ]]; then\n sed -i -e '/ConvertLayoutOpToLLVM.cpp/a ConvertLayoutOpToLLVM/SharedToSparseDotOperand.cpp' third_party/nvidia/lib/TritonNVIDIAGPUToLLVM/CMakeLists.txt\n sed -i -e '/DotOpToLLVM.cpp/a DotOpToLLVM/Sparse.cpp' third_party/nvidia/lib/TritonNVIDIAGPUToLLVM/CMakeLists.txt\nfi\n# Use clang to match Google etc.\nLLVM_INCLUDE_DIRS=/opt/llvm-build/include \\\n LLVM_LIBRARY_DIR=/opt/llvm-build/lib \\\n LLVM_SYSPATH=/opt/llvm-build \\\n TRITON_BUILD_WITH_CLANG_LLD=1 \\\n pip wheel --verbose --wheel-dir=dist/ python/\n# Clean up the wheel build directory so it doesn't end up bloating the container\nrm -rf python/build\n# Make the layer for the *current* step smaller, so it is more likely to stay\n# resident in the Docker cache\ncp -r /opt/checkout/external/triton /opt/triton-copy\nrm -rf /opt/checkout /opt/llvm-build /root/.cache\nEOF" did not complete successfully: exit code: 1