From 980796cd075cb8e2591352a725e8adb668152aab Mon Sep 17 00:00:00 2001 From: "Li, Jiang" Date: Mon, 8 Jun 2026 19:10:06 +0800 Subject: [PATCH] [CI/Build][CPU] Fix flaky CI image build failure and unexpected warnings (#44852) Signed-off-by: jiang1.li --- docker/Dockerfile.cpu | 6 +++--- vllm/_custom_ops.py | 3 ++- 2 files changed, 5 insertions(+), 4 deletions(-) diff --git a/docker/Dockerfile.cpu b/docker/Dockerfile.cpu index e185c00cb2..02af3fd3c3 100644 --- a/docker/Dockerfile.cpu +++ b/docker/Dockerfile.cpu @@ -118,9 +118,9 @@ ENV CARGO_BUILD_JOBS=4 # Build the release binary. Cache cargo registry/git and target/, but copy the # binary out of the target/ cache mount so it persists into the image layer # for later COPY --from=rust-build. -RUN --mount=type=cache,target=/root/.cargo/registry \ - --mount=type=cache,target=/root/.cargo/git \ - --mount=type=cache,target=/workspace/rust/target \ +RUN --mount=type=cache,target=/root/.cargo/registry,sharing=locked \ + --mount=type=cache,target=/root/.cargo/git,sharing=locked \ + --mount=type=cache,target=/workspace/rust/target,sharing=locked \ VLLM_RS_TARGET_PATH=/workspace/vllm-rs bash build_rust.sh ######################### BUILD IMAGE ######################### diff --git a/vllm/_custom_ops.py b/vllm/_custom_ops.py index 2c0684c27f..3bac6972f1 100644 --- a/vllm/_custom_ops.py +++ b/vllm/_custom_ops.py @@ -3537,10 +3537,11 @@ class CPUDNNLGEMMHandler: self.handler_tensor: torch.Tensor | None = None self.n = -1 self.k = -1 + self.dtor = torch.ops._C.release_dnnl_matmul_handler def __del__(self): if self.handler_tensor is not None: - torch.ops._C.release_dnnl_matmul_handler(self.handler_tensor.item()) + self.dtor(self.handler_tensor.item()) _supports_onednn = bool(hasattr(torch.ops._C, "create_onednn_mm_handler"))