versions.env 3.5 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667
  1. # Authoritative build inputs — GLM-5.3-Flash r12 (jovian-judgement) clean build.
  2. # Every entry is a tag or exact version — no :latest, no main, no ranges.
  3. # Pattern: solar-open2-nvfp4 (raw CUDA base, staged wheel builds, ccache/uv
  4. # buildx caches) — content r12-specific per the rtx6kpro source contract.
  5. # CUDA base (aarch64; digest-pinned, tested by solar-open2-nvfp4)
  6. CUDA_BASE_IMAGE=nvidia/cuda:13.2.0-devel-ubuntu24.04
  7. CUDA_BASE_DIGEST=sha256:a5b6256e470196fc1d5f8f62139d57d3662867746dfe1cb352d7652024047020
  8. # Python / uv
  9. UV_VERSION=0.11.28
  10. PYPI_INDEX_URL=https://pypi.org/simple
  11. PYTORCH_INDEX_URL=https://download.pytorch.org/whl/cu130
  12. # Torch stack — the fork's requirements/cuda.txt pins (r12 era, cu130 aarch64)
  13. TORCH_VERSION=2.13.0
  14. TORCHVISION_VERSION=0.28.0
  15. TORCHAUDIO_VERSION=2.11.0
  16. # FlashInfer — the fork's cuda.txt uses flashinfer-python==0.6.17 (PyPI);
  17. # flashinfer-cubin is NOT a build dep (setup.py skips it; the runtime
  18. # fetches cubins when the package is absent).
  19. FLASHINFER_VERSION=0.6.17
  20. # vLLM — local-inference-lab fork, dev/jovian-judgement base + PR heads.
  21. #
  22. # r18 composition (2026-09-05): branch tip f564dffe9a66 — supersedes the r17
  23. # base 9c4dd054 (which itself superseded r12's 54f6e9826c + PR set 550-557;
  24. # do NOT re-merge 550-557, they conflict). The r17 PR set 579/580/581/586 is
  25. # now merged into the tip; only 582 is still open, pinned by head SHA.
  26. # New in this window (9c4dd054..f564dffe + b12x below): RoCEnante one-shot
  27. # RoCE collectives (PR #597; env VLLM_ENABLE_ROCE_ALLREDUCE, off by default),
  28. # B12X NVFP4 KV cache (fp8_ds_mla / nvfp4_ds_mla), DFlash split-KV paged
  29. # decode attention, L2 weight-prefetch decode, DSpark input-budget split,
  30. # spec-decode KV-connector + prefix-cache-block fixes, GLM tool-delimiter
  31. # parser hardening, prefill fairness engines, KDA prefill backend,
  32. # FlashKDA stable-inverse fix. No requirements/setup.py drift vs r17.
  33. # Staged tree on the build host: sm121/r18 @ 725e36d3189cc7cbd8c27a16323c4b9643fd8cf2
  34. # (base + pr582 merge + staged-cutlass cmake patch; see /tmp/vllm-src.tar.gz).
  35. VLLM_REPO=https://github.com/local-inference-lab/vllm.git
  36. VLLM_BASE_REF=f564dffe9a6625ae3d87706530ce35974f1d2989
  37. VLLM_PR_HEADS="582:4df4cf5e123b55aab2fbcb310468b076995e9410"
  38. # B12X — master + PR heads (pure-python wheel; CuTe-DSL JIT kernels).
  39. # r18: master tip 613cfd9b (supersedes r17 base 2bbab479; PR 290 bf16 PCIe
  40. # two-shot now merged into it). Still open: 284 (SM120 mhc decode split),
  41. # pinned by head SHA. New in window: RoCEnante one-shot RoCE all-reduce/
  42. # all-gather (#295, both-rail striping, DSV4 graph collectives), GLM_NEXT
  43. # NVFP4 KV cache (PR #309), KDA prefill extension (full op surface),
  44. # dense-GEMM/MoE A16 activation-precision selection on SM121/GB10 (relevant
  45. # to the linear-b12x quality toxin — re-bisect with this build), QSA large-
  46. # prefill + paged-CuTe specialization, DSA indexer plan-owned lifecycle.
  47. # Staged tree on the build host: sm121/r18 @ c594d447d3129e717f8aa0a7c56d7b777912bec3
  48. # (base + pr284 merge; see /tmp/b12x-src.tar.gz).
  49. B12X_REPO=https://github.com/local-inference-lab/b12x.git
  50. B12X_BASE_REF=613cfd9bbc542e2a7a9e5e216d1f62a8becda87f
  51. B12X_PRS="284"
  52. # DeepGEMM — fetched by vllm's cmake; pre-cloned (raw-SHA fetches of a
  53. # non-default-branch tip 404 over GitHub's git protocol)
  54. DEEPGEMM_REPO=https://github.com/deepseek-ai/DeepGEMM.git
  55. DEEPGEMM_BRANCH=nv_dev
  56. DEEPGEMM_REF=8b1392b978f5a03c828dd1711090d7fb50958b8a
  57. # Target GPU architecture (DGX Spark = sm_121a)
  58. TORCH_CUDA_ARCH_LIST=12.1a
  59. BUILD_JOBS=4