Install additional dependencies

Author: zacblazic

Dockerfile.arm64

@@ -1,13 +1,20 @@
 FROM python:3.12-slim AS triton-builder
 
 RUN apt-get update -qq && \
-    apt-get install -qq -y git && \
+    apt-get install -qq -y \
+        build-essential \
+        clang \
+        cmake \
+        git \
+        lld \
+        llvm && \
     rm -rf /var/lib/apt/lists/*
 
 RUN git clone https://github.com/triton-lang/triton.git /tmp/triton \
       --depth 1 \
       --branch v3.2.0
 
+ARG TRITON_BUILD_WITH_CLANG_LLD=true
 RUN --mount=type=cache,target=/root/.cache/pip \
     cd /tmp/triton && \
     pip install ninja cmake wheel pybind11 && \