Skip to content

Commit ca1fdc4

Browse files
divakar-amdrasmith
authored andcommitted
[bugfix] moe tuning. rm is_navi() (vllm-project#12273)
Signed-off-by: Divakar Verma <[email protected]>
1 parent e5d3dff commit ca1fdc4

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)