versions.env 3.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566
  1. # Authoritative build inputs — karmic-kraken (local-inference-lab vLLM fork)
  2. # clean build. Provenance: dev/karmic-kraken tip 636afde9 (2026-10-08), which
  3. # takes B12X from FlashInfer 0.7.1 (the standalone b12x repo is archived).
  4. # Supersedes the sparkring dev-20261001-kraken provenance (vllm ab86b7073,
  5. # b12x e4084d2e, flashinfer 36ad74ad).
  6. #
  7. # Every entry is a tag, exact version, branch-name at a pinned commit, or a
  8. # verified tree hash — no :latest, no ranges, no floating tips.
  9. #
  10. # Reproducibility contract (clone-at-pin, NO staged tarballs): the Dockerfile
  11. # clones every source at the pins below inside the build and GATES each one
  12. # (HEAD commit + tree hash must match, pinned deps must appear verbatim in the
  13. # tree's cmake). A fresh clone of this repository + `bash build.sh` reproduces
  14. # the image.
  15. # CUDA base (aarch64; the sparkring image uses CUDA 13.4.2)
  16. CUDA_BASE_IMAGE=nvidia/cuda:13.4.2-devel-ubuntu24.04
  17. CUDA_BASE_DIGEST=sha256:d44d6dc249c2c8330d0b380c595a514f0e86137fcc717fe7a4389a25720ade37
  18. # Python / uv
  19. UV_VERSION=0.11.28
  20. PYPI_INDEX_URL=https://pypi.org/simple
  21. PYTORCH_INDEX_URL=https://download.pytorch.org/whl/cu130
  22. # apt mirror for the apt-base stage: current = live ports.ubuntu.com
  23. APT_SOURCE=current
  24. # DeepGEMM — cloned in-build; pin gated in fetch-deps.sh against the tree's
  25. # own deepgemm.cmake declaration.
  26. DEEPGEMM_REPO=https://github.com/deepseek-ai/DeepGEMM.git
  27. DEEPGEMM_BRANCH=nv_dev
  28. DEEPGEMM_REF=8b1392b978f5a03c828dd1711090d7fb50958b8a
  29. # Torch stack — the fork's requirements/cuda.txt pins
  30. TORCH_VERSION=2.13.0
  31. TORCHVISION_VERSION=0.28.0
  32. TORCHAUDIO_VERSION=2.11.0
  33. # FlashInfer — 0.7.1 line (the karmic-kraken tip's pin; B12X now ships inside it).
  34. # No FINAL 0.7.1 wheel exists yet (the line is at 0.7.1rc5, and 0.7.2rc1 is
  35. # already out) — so we pin the latest 0.7.1-line wheel, 0.7.1rc5, from the
  36. # flashinfer.ai/whl index. FlashInfer is NOT a core vllm-wheel dependency
  37. # (only the `vllm[b12x]` extra pulls it), and we install it separately in
  38. # torch-base, so the rc pin is safe. Its commit is v0.7.1rc5 = 9a39cddc.
  39. FLASHINFER_VERSION=0.7.1rc5
  40. FLASHINFER_COMMIT=9a39cddccee9854a00b2824e64c937d37ad316d7
  41. # vLLM — karmic-kraken fork (local-inference-lab/vllm@dev/karmic-kraken)
  42. VLLM_REPO=https://github.com/local-inference-lab/vllm.git
  43. VLLM_BRANCH=dev/karmic-kraken
  44. VLLM_PIN=636afde98300e1492d7bfc5665f8dcfaec8b706e
  45. VLLM_PRETEND_VERSION=0.1.dev21808+g636afde98.d20261008
  46. # B12X — ships inside flashinfer-python 0.7.1 (flashinfer/experimental/b12x plus
  47. # the top-level `b12x` import package). The standalone lukealonso/b12x repo is
  48. # archived; a separate b12x wheel must NOT be installed next to flashinfer 0.7.1
  49. # (they own the same namespace). The Dockerfile's b12x-clone/builder stages are
  50. # dropped accordingly — no B12X_* build-args are consumed anymore.
  51. # NCCL — the sparkring image's pin (2.32.3, rev 12df1a11)
  52. NCCL_VERSION=2.32.3
  53. NCCL_REVISION=12df1a11afad322be5a204a2db890161cbf8131d
  54. # Target GPU architecture (DGX Spark = sm_121a)
  55. TORCH_CUDA_ARCH_LIST=12.1a
  56. BUILD_JOBS=16