Files
amd-strix-halo-vllm-toolboxes/scripts/01-rocm-env-for-triton.sh
T

8 lines
285 B
Bash
Raw Normal View History

2025-11-30 14:57:37 +00:00
# Required for Strix Halo / RDNA3.5 on vLLM
export TORCH_ROCM_AOTRITON_ENABLE_EXPERIMENTAL=1
export FLASH_ATTENTION_TRITON_AMD_ENABLE="TRUE"
export VLLM_TARGET_DEVICE=rocm
export VLLM_USE_MMAP=0
export VLLM_ROCM_USE_AITER=0
export VLLM_ROCM_USE_AITER_MOE=0
export VLLM_USE_TRITON_AWQ=1