Skip to content

Commit 2acba47

Browse files
authored
[bugfix] moe tuning. rm is_navi() (#12273)
Signed-off-by: Divakar Verma <divakar.verma@amd.com>
1 parent 9c485d9 commit 2acba47

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

benchmarks/kernels/benchmark_moe.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,10 +12,10 @@
1212

1313
from vllm.model_executor.layers.fused_moe.fused_moe import *
1414
from vllm.platforms import current_platform
15-
from vllm.utils import FlexibleArgumentParser, is_navi
15+
from vllm.utils import FlexibleArgumentParser
1616

1717
FP8_DTYPE = torch.float8_e4m3fnuz if current_platform.is_rocm(
18-
) and not is_navi() else torch.float8_e4m3fn
18+
) else torch.float8_e4m3fn
1919

2020

2121
class BenchmarkConfig(TypedDict):

0 commit comments

Comments
 (0)