software-engineering/spark-vllm-docker
2
0
Fork 0
Code Issues Actions Packages Projects Releases Wiki Activity

Changed CUTLASS ref for mxfp4 build

Browse Source
This commit is contained in:
eugr
2026-04-01 08:58:31 -07:00
parent a770865834
commit ae25d64ac0
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
Dockerfile.mxfp4
View File

@@ -98,7 +98,7 @@ ARG FLASHINFER_REPO=https://github.com/christopherowen/flashinfer.git
ARG CUTLASS_REPO=https://github.com/christopherowen/cutlass.git ARG CUTLASS_REPO=https://github.com/christopherowen/cutlass.git
ARG FLASHINFER_SHA=f349e52496a72a00d8c4ac02c7a1e38523ff7194 ARG FLASHINFER_SHA=f349e52496a72a00d8c4ac02c7a1e38523ff7194
ARG CUTLASS_SHA=c7516ad20f3d022fdbc93e9468643bf3b577e02c ARG CUTLASS_SHA=fede53000a962b46e05bafe0c86311778caeb380
RUN --mount=type=cache,id=uv-cache,target=/root/.cache/uv \ RUN --mount=type=cache,id=uv-cache,target=/root/.cache/uv \
uv pip install nvidia-nvshmem-cu13 "apache-tvm-ffi<0.2" uv pip install nvidia-nvshmem-cu13 "apache-tvm-ffi<0.2"