From fe244e3f8baa984050be6e5c852f70be44bfe716 Mon Sep 17 00:00:00 2001 From: Sayak Paul Date: Fri, 21 Aug 2026 06:14:19 +0000 Subject: [PATCH] compilation failure fixes --- docker/diffusers-pytorch-cuda/Dockerfile | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/docker/diffusers-pytorch-cuda/Dockerfile b/docker/diffusers-pytorch-cuda/Dockerfile index 345bef4648b8..8052022eb236 100644 --- a/docker/diffusers-pytorch-cuda/Dockerfile +++ b/docker/diffusers-pytorch-cuda/Dockerfile @@ -10,11 +10,14 @@ RUN apt-get -y update \ && add-apt-repository ppa:deadsnakes/ppa && \ apt-get update -# cuda-cudart-dev provides the CUDA headers (cuda.h etc.) AOT-inductor compilation needs; -# the runtime base image only ships the libraries. +# The runtime base image only ships the CUDA libraries, but AOT-inductor compiles a C++ +# wrapper against the CUDA headers. cuda-cudart-dev provides cuda.h/cuda_runtime_api.h, and +# cuda-nvcc pulls in the crt/ headers (crt/host_defines.h etc.) that cuda_runtime_api.h +# includes in turn -- both are needed or the AOT tests fail with a CppCompileError. RUN apt install -y bash \ build-essential \ cuda-cudart-dev-12-9 \ + cuda-nvcc-12-9 \ git \ git-lfs \ curl \