From c1cad896f4ddc449daac11911ea1bf59089a7e9a Mon Sep 17 00:00:00 2001
From: risemeup1 <62429225+risemeup1@users.noreply.github.com>
Date: Mon, 10 Apr 2023 10:33:13 +0800
Subject: [PATCH] optimize setup.py (#52621)

* optimize setup.py

* add ninja
---
 setup.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/setup.py b/setup.py
index d1f125e6b4f..daa9dbd5cc6 100644
--- a/setup.py
+++ b/setup.py
@@ -685,6 +685,11 @@ def options_process(args, build_options):
 def get_cmake_generator():
     if os.getenv("GENERATOR"):
         cmake_generator = os.getenv("GENERATOR")
+        if os.system('ninja --version') == 0:
+            print("Ninja has been installed,use ninja to compile Paddle now.")
+        else:
+            print("Ninja has not been installed,install it now.")
+            os.system('python -m pip install ninja')
     else:
         cmake_generator = "Unix Makefiles"
     return cmake_generator
-- 
GitLab