[Feat] allow using aclgraph in ray backend

Signed-off-by: WithHades <244036962@qq.com>

Author: WithHades

tests/ut/test_platform.py

@@ -351,34 +351,6 @@ def test_check_and_update_config_unsupported_cudagraph_mode(
                 CUDAGraphMode.NONE,
             )
 
-    @patch("vllm_ascend.utils.is_310p", return_value=False)
-    @patch("vllm_ascend.ascend_config.check_ascend_config")
-    @patch("vllm_ascend.ascend_config.init_ascend_config")
-    def test_check_and_update_config_disable_aclgraph_when_ray_enabled(
-            self, mock_init_ascend, mock_check_ascend, mock_is_310p):
-        mock_init_ascend.return_value = self.mock_ascend_config
-        self.mock_vllm_config.model_config.enforce_eager = False
-        self.mock_vllm_config.compilation_config.level = CompilationLevel.PIECEWISE
-        self.mock_vllm_config.parallel_config.distributed_executor_backend = "ray"
-
-        with self.assertLogs(logger="vllm", level="WARNING") as cm:
-            from vllm_ascend import platform
-
-            importlib.reload(platform)
-            self.platform.check_and_update_config(self.mock_vllm_config)
-            print(30 * "=", f"cm.output: {cm.output}")
-            self.assertTrue(
-                "Ray distributed executor backend is not compatible with ACL Graph mode"
-                in cm.output[0])
-            self.assertEqual(
-                self.mock_vllm_config.compilation_config.level,
-                CompilationLevel.NO_COMPILATION,
-            )
-            self.assertEqual(
-                self.mock_vllm_config.compilation_config.cudagraph_mode,
-                CUDAGraphMode.NONE,
-            )
-
     @patch("vllm_ascend.utils.is_310p", return_value=False)
     @patch("vllm_ascend.ascend_config.check_ascend_config")
     @patch("vllm_ascend.ascend_config.init_ascend_config")

vllm_ascend/platform.py

@@ -171,12 +171,6 @@ def check_and_update_config(cls, vllm_config: VllmConfig) -> None:
             )
             compilation_config.cudagraph_mode = CUDAGraphMode.NONE
 
-        if parallel_config.distributed_executor_backend == "ray":
-            logger.warning(
-                "Ray distributed executor backend is not compatible with ACL Graph mode "
-                "right now. Setting CUDAGraphMode to NONE")
-            compilation_config.cudagraph_mode = CUDAGraphMode.NONE
-
         # set cudaprah sizes before extending `compilation_config.splitting_ops`
         vllm_config._set_cudagraph_sizes()