xx

Author: wangxiaoxin-sherie

vllm_ascend/platform.py

@@ -178,11 +178,7 @@ def check_and_update_config(cls, vllm_config: VllmConfig) -> None:
 
         compilation_config.cudagraph_num_of_warmups = 1
 
-        if compilation_config.level == CompilationLevel.PIECEWISE:
-            logger.warning(
-                "NEW  NPU does not support %s compilation level. Setting CUDAGraphMode to NONE",
-                compilation_config.level)
-        elif compilation_config.level not in [
+        if compilation_config.level not in [
                 CompilationLevel.NO_COMPILATION, CompilationLevel.PIECEWISE
         ]:
             logger.warning(