remove exclusive_mode

Author: shihaobai

lightllm/server/api_start.py

@@ -65,10 +65,8 @@ def normal_or_p_d_start(args):
     set_unique_server_name(args)
 
     if args.enable_mps:
-        from lightllm.utils.device_utils import enable_mps, set_gpu_exclusive_mode
+        from lightllm.utils.device_utils import enable_mps
 
-        for i in range(args.tp):
-            set_gpu_exclusive_mode(gpu_index=i)
         enable_mps()
 
     if args.run_mode not in ["normal", "prefill", "decode"]:

lightllm/utils/start_utils.py

@@ -62,13 +62,10 @@ def kill_recursive(proc):
 
         # recover the gpu compute mode
         is_enable_mps = get_env_start_args().enable_mps
-        world_size = get_env_start_args().tp
         if is_enable_mps:
-            from lightllm.utils.device_utils import stop_mps, set_gpu_default_mode
+            from lightllm.utils.device_utils import stop_mps
 
             stop_mps()
-            for i in range(world_size):
-                set_gpu_default_mode(gpu_index=i)
         logger.info("All processes terminated gracefully.")