versions.env 1.6 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556
  1. # versions.env - authoritative build inputs for GLM-5.2 NF3-Hybrid
  2. # Every entry is a tag or exact version - no :latest, no main, no ranges.
  3. CUDA_BASE_IMAGE=nvidia/cuda:13.2.0-devel-ubuntu24.04
  4. CUDA_BASE_DIGEST=sha256:a5b6256e470196fc1d5f8f62139d57d3662867746dfe1cb352d7652024047020
  5. GB10_BUILD=1
  6. # Python / uv
  7. UV_VERSION=0.11.28
  8. # PyTorch stack (cu130 index works on CUDA 13.2 for aarch64)
  9. TORCH_VERSION=2.11.0
  10. TORCHVISION_VERSION=0.26.0
  11. TORCHAUDIO_VERSION=2.11.0
  12. TRITON_VERSION=3.6.0
  13. PYTORCH_INDEX_URL=https://download.pytorch.org/whl/cu130
  14. PYPI_INDEX_URL=https://pypi.org/simple
  15. # CUDA-side companion packages
  16. NVSHMEM_VERSION=3.4.5
  17. TVM_FFI_VERSION=0.1.9
  18. TILELANG_VERSION=0.1.9
  19. NUMBA_VERSION=0.65.0
  20. # NCCL
  21. NCCL_REPO=https://github.com/NVIDIA/nccl.git
  22. NCCL_REF=v2.30.7-1
  23. # vLLM (local-inference-lab fork with b12x, DCP, MTP support)
  24. VLLM_REPO=https://github.com/local-inference-lab/vllm.git
  25. VLLM_REF=codex/dcp-globaltopk-sharddraft-defaults-20260622
  26. VLLM_COMMIT=e232d262369b8c918cf478a7a96a0fcf8127cf65
  27. # b12x (NVFP4 kernel library)
  28. B12X_REPO=https://github.com/voipmonitor/b12x.git
  29. B12X_REF=f3686b55
  30. B12X_COMMIT=f3686b555d639823b276c2080f173145eed7f007
  31. # FlashInfer
  32. FLASHINFER_REPO=https://github.com/flashinfer-ai/flashinfer.git
  33. FLASHINFER_REF=v0.6.12
  34. # cutlass-dsl (4.5.3 for firmware 580.159.03 compat)
  35. CUTLASS_DSL_VERSION=4.5.3
  36. # Runtime lock seeds
  37. RAY_VERSION=2.56.0
  38. FASTSAFETENSORS_VERSION=0.2.2
  39. INSTANTTENSOR_VERSION=0.1.8
  40. BITSANDBYTES_VERSION=0.49.2
  41. ACCELERATE_VERSION=1.14.0
  42. # Target GPU architecture (DGX Spark = sm_121a)
  43. TORCH_CUDA_ARCH_LIST=12.1a
  44. FLASHINFER_CUDA_ARCH_LIST=12.1a