File tree Expand file tree Collapse file tree 1 file changed +18
-0
lines changed
Expand file tree Collapse file tree 1 file changed +18
-0
lines changed Original file line number Diff line number Diff line change 22# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
33"""Pytest configuration for vLLM multimodal tests."""
44
5+ import os
56import warnings
67
78import torch
89
910from vllm .platforms import current_platform
1011
1112
13+ def pytest_configure (config ):
14+ """Early ROCm configuration that must happen before test collection."""
15+ if not current_platform .is_rocm ():
16+ return
17+
18+ # Disable skinny GEMM on ROCm to avoid non-deterministic results
19+ # from atomic reductions in wvSplitKrc kernel.
20+ # See: https://github.com/vllm-project/vllm/pull/33493#issuecomment-3906083975
21+ os .environ ["VLLM_ROCM_USE_SKINNY_GEMM" ] = "0"
22+ warnings .warn (
23+ "ROCm: Set VLLM_ROCM_USE_SKINNY_GEMM=0 to avoid non-deterministic "
24+ "results from skinny GEMM atomic reductions" ,
25+ UserWarning ,
26+ stacklevel = 1 ,
27+ )
28+
29+
1230def pytest_collection_modifyitems (config , items ):
1331 """Configure ROCm-specific settings based on collected tests."""
1432 if not current_platform .is_rocm ():
You can’t perform that action at this time.
0 commit comments