versions.env 2.6 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970
  1. # versions.env - authoritative build inputs for DeepSeek-V4-Flash DSpark (from source)
  2. # EXPERIMENTAL: builds vLLM + b12x from the local-inference-lab "eldritch" fork at the
  3. # newest DS4/DSpark branch, for DGX Spark (aarch64 / sm_121a). Every entry pinned.
  4. #
  5. # Reference stack inputs come from the fork's own tools/spark/versions.env (the
  6. # authoritative pairing for the pinned branch) plus the GLM-5.2 NF3-Hybrid harness
  7. # this directory is derived from (CUDA base, apt locks, checksums).
  8. # --- CUDA base ---
  9. CUDA_BASE_IMAGE=nvidia/cuda:13.2.0-devel-ubuntu24.04
  10. CUDA_BASE_DIGEST=sha256:a5b6256e470196fc1d5f8f62139d57d3662867746dfe1cb352d7652024047020
  11. GB10_BUILD=1
  12. # --- Python / uv ---
  13. UV_VERSION=0.11.28
  14. # --- PyTorch stack (cu130 index works on CUDA 13.2 for aarch64) ---
  15. # NOTE: the newest DS4 fork line moved to torch 2.12.0. The python-build.txt lock
  16. # MUST be regenerated against these pins (see gen-locks.sh) before building.
  17. TORCH_VERSION=2.12.0
  18. TORCHVISION_VERSION=0.27.0
  19. TORCHAUDIO_VERSION=2.12.0
  20. TRITON_VERSION=3.7.0
  21. PYTORCH_INDEX_URL=https://download.pytorch.org/whl/cu130
  22. PYPI_INDEX_URL=https://pypi.org/simple
  23. # --- CUDA-side companion packages ---
  24. NVSHMEM_VERSION=3.4.5
  25. TVM_FFI_VERSION=0.1.11
  26. TILELANG_VERSION=0.1.9
  27. NUMBA_VERSION=0.65.0
  28. # --- NCCL (pinned from tools/spark/versions.env) ---
  29. NCCL_REPO=https://github.com/NVIDIA/nccl.git
  30. NCCL_REF=5067397c2676d5aed50042fc39e5c8ee96eb0027
  31. # --- vLLM (local-inference-lab b12x fork, newest DS4/DSpark line) ---
  32. VLLM_REPO=https://github.com/local-inference-lab/vllm.git
  33. VLLM_REF=feat/gg-ds4-0731-launcher-20260731
  34. VLLM_COMMIT=e63a190ee1b48d051ccebce485de343dbe2b3505
  35. # --- b12x (NVFP4 kernel library). Pin matches the fork's eldritch-fullstack line;
  36. # recheck against the branch's tools/spark/versions.env if it changes. ---
  37. B12X_REPO=https://github.com/voipmonitor/b12x.git
  38. B12X_REF=284a2eae83754ee1abd31c37b9ca66b68e20b8a8
  39. B12X_COMMIT=284a2eae83754ee1abd31c37b9ca66b68e20b8a8
  40. # --- FlashInfer (pinned from tools/spark/versions.env) ---
  41. FLASHINFER_REPO=https://github.com/flashinfer-ai/flashinfer.git
  42. FLASHINFER_REF=ce40d25a4cd05b2f28c935b99c9ea5091b4e64d2
  43. FLASHINFER_VERSION=0.6.15
  44. # --- DeepGEMM (informational; consumed as a b12x/flashinfer build dep) ---
  45. DEEPGEMM_REF=a6b593d2826719dcf4892609af7b84ee23aaf32a
  46. # --- cutlass-dsl (4.6.0 — matches the fork's requirements/cuda.txt pin) ---
  47. CUTLASS_DSL_VERSION=4.6.0
  48. # --- Runtime lock seeds ---
  49. RAY_VERSION=2.56.0
  50. FASTSAFETENSORS_VERSION=0.3.2
  51. INSTANTTENSOR_VERSION=0.1.9
  52. BITSANDBYTES_VERSION=0.49.2
  53. ACCELERATE_VERSION=1.14.0
  54. # --- Target GPU architecture (DGX Spark = sm_121a) ---
  55. TORCH_CUDA_ARCH_LIST=12.1a
  56. FLASHINFER_CUDA_ARCH_LIST=12.1a
  57. CUTE_DSL_ARCH=sm_121a