21 lines
618 B
Python
21 lines
618 B
Python
|
#!/usr/bin/env python3
|
||
|
|
# SPDX-License-Identifier: Apache-2.0
|
||
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
||
|
|
"""Check that vllm.platforms import does not initialize CUDA."""
|
||
|
|
|
||
|
|
import os
|
||
|
|
|
||
|
|
for key in ["CUDA_VISIBLE_DEVICES", "HIP_VISIBLE_DEVICES", "ROCR_VISIBLE_DEVICES"]:
|
||
|
|
os.environ.pop(key, None)
|
||
|
|
|
||
|
|
import torch # noqa: E402
|
||
|
|
|
||
|
|
assert not torch.cuda.is_initialized(), "CUDA initialized before import"
|
||
|
|
|
||
|
|
from vllm.platforms import current_platform # noqa: E402
|
||
|
|
|
||
|
|
assert not torch.cuda.is_initialized(), (
|
||
|
|
f"CUDA was initialized during vllm.platforms import on {current_platform}"
|
||
|
|
)
|
||
|
|
print("OK")
|