dispram.yaml 1.1 KB

1234567891011121314151617181920212223
  1. # The GB10's display carveout as KV cache. The firmware reserves 2 GiB for a
  2. # display that the driver never uses on this chip. kindling-spark-os
  3. # (github.com/kindlingai/kindling-spark-os) runs dispramd, which lends that
  4. # range to CUDA processes. This file mounts dispramd's socket and the host's
  5. # copy of its vLLM plugin, so the client always matches the daemon. The plugin
  6. # adds what dispramd can lend (1.99 GiB) to the pinned KV pool and maps the
  7. # pool's tail onto it.
  8. #
  9. # kindling-spark-os's 64 KiB kernel also returns ~2 GiB of RAM, so at TP=4
  10. # this raises the pinned pool by 2 GiB (KV_EXTRA_MEMORY). TP=2 has no room for
  11. # it, and TP=3 and 6 are unmeasured.
  12. #
  13. # ./glm53 adds this file by itself when /run/dispram/dispram.sock exists.
  14. # DISPRAM=0 leaves it out. Every rank should have it: vLLM sizes the pool by
  15. # the smallest rank, so one box without dispramd cancels the gain.
  16. services:
  17. glm53:
  18. volumes:
  19. - /run/dispram:/run/dispram
  20. - /opt/kindling/dispram/python:/opt/dispram:ro
  21. environment:
  22. - PYTHONPATH=/opt/dispram
  23. - KV_EXTRA_MEMORY=${KV_EXTRA_MEMORY:-2147483648}