[bugfix] moe tuning. rm is_navi() (#12273)

Signed-off-by: Divakar Verma <[email protected]>
vllm-project · Jan 21, 2025 · 2acba47 · 2acba47
1 parent 9c485d9
commit 2acba47
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/benchmarks/kernels/benchmark_moe.py b/benchmarks/kernels/benchmark_moe.py
@@ -12,10 +12,10 @@
 
 from vllm.model_executor.layers.fused_moe.fused_moe import *
 from vllm.platforms import current_platform
-from vllm.utils import FlexibleArgumentParser, is_navi
+from vllm.utils import FlexibleArgumentParser
 
 FP8_DTYPE = torch.float8_e4m3fnuz if current_platform.is_rocm(
-) and not is_navi() else torch.float8_e4m3fn
+) else torch.float8_e4m3fn
 
 
 class BenchmarkConfig(TypedDict):