e37b4d283b
CCCL buddy_allocator.cu teaches: control memory block fragmentation at the allocator level. Sub168 OOM trace shows 'max_split_size_mb' suggestion. Adding PYTORCH_CUDA_ALLOC_CONF=max_split_size_mb:512 prevents PyTorch memory fragmentation that caused Sub168's final OOM. OMP_NUM_THREADS=1 matches Sub168 docker log: 'Reducing Torch parallelism from 64 threads to 1' CCCL device_reduce policy_selector pattern: hardware-adaptive params through environment, not code changes. computility-run.yaml env vars are the serving-safe equivalent of CCCL policy_selector.
58 lines
1.5 KiB
YAML
58 lines
1.5 KiB
YAML
concurrency: 1
|
|
command:
|
|
- python3
|
|
- -m
|
|
- vllm.entrypoints.openai.api_server
|
|
- --model
|
|
- /model
|
|
- --served-model-name
|
|
- llm
|
|
- --max-model-len
|
|
- '256000'
|
|
- --gpu-memory-utilization
|
|
- '0.95'
|
|
- --trust-remote-code
|
|
- -tp
|
|
- '4'
|
|
- --max-num-seqs
|
|
- '2'
|
|
- --max-num-batched-tokens
|
|
- '4096'
|
|
- --disable-log-requests
|
|
- --disable-frontend-multiprocessing
|
|
- --enforce-eager
|
|
- --enable-auto-tool-choice
|
|
- --tool-call-parser
|
|
- qwen3_coder
|
|
- --reasoning-parser
|
|
- qwen3
|
|
- --enable-prefix-caching
|
|
- --enable-chunked-prefill
|
|
- --dtype
|
|
- half
|
|
env:
|
|
- name: VLLM_ENGINE_ITERATION_TIMEOUT_S
|
|
value: '3600'
|
|
- name: VLLM_ATTENTION_BACKEND
|
|
value: XFORMERS
|
|
- name: ENABLE_CUSTOM_IPC
|
|
value: '1'
|
|
- name: PYTHONPATH
|
|
value: /usr/local/corex/lib/python3/dist-packages:/usr/local/corex/lib64/python3/dist-packages
|
|
- name: LD_LIBRARY_PATH
|
|
value: /usr/local/corex/lib64:/usr/local/openmpi/lib
|
|
- name: VLLM_COREX_FA2_LIBRARY
|
|
value: /usr/local/corex/lib64/libcorex_fa2.so
|
|
- name: VLLM_COREX_GDN_LIBRARY
|
|
value: /usr/local/corex/lib64/libcorex_gdn.so
|
|
- name: VLLM_COREX_MOE_LIBRARY
|
|
value: /usr/local/corex/lib64/libcorex_moe.so
|
|
- name: VLLM_REQUEST_METRICS_FILE
|
|
value: /tmp/vllm-request-metrics.jsonl
|
|
- name: VLLM_CACHE_BLOCK_SIZE
|
|
value: '16'
|
|
- name: PYTORCH_CUDA_ALLOC_CONF
|
|
value: max_split_size_mb:512
|
|
- name: OMP_NUM_THREADS
|
|
value: '1'
|