Unverified Commit 98ff0429 authored by rasmith's avatar rasmith Committed by GitHub
Browse files

[CI][BugFix][AMD] Don't set VLLM_ROCM_USE_AITER anymore in...


[CI][BugFix][AMD] Don't set VLLM_ROCM_USE_AITER anymore in test_rocm_aiter_topk since its not necessary (#36996)
Signed-off-by: default avatarRandall Smith <Randall.Smith@amd.com>
parent b55156ea
...@@ -10,7 +10,6 @@ ...@@ -10,7 +10,6 @@
# and the platform is not ROCm. # and the platform is not ROCm.
import importlib.util import importlib.util
import os
import pytest import pytest
import torch import torch
...@@ -20,9 +19,6 @@ from vllm.platforms import current_platform ...@@ -20,9 +19,6 @@ from vllm.platforms import current_platform
if not current_platform.is_rocm(): if not current_platform.is_rocm():
pytest.skip("This test can only run on ROCm.", allow_module_level=True) pytest.skip("This test can only run on ROCm.", allow_module_level=True)
# This environment variable must be set so ops will be registered.
os.environ["VLLM_ROCM_USE_AITER"] = "1"
# this import statement is needed to ensure the ops are registered # this import statement is needed to ensure the ops are registered
import vllm.model_executor.layers.fused_moe.rocm_aiter_fused_moe # noqa: F401 import vllm.model_executor.layers.fused_moe.rocm_aiter_fused_moe # noqa: F401
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment