From 89f6e1db5e1fa6720b940fb2be8bf28cdde48776 Mon Sep 17 00:00:00 2001
From: Ian Bearman <ianb@microsoft.com>
Date: Sun, 9 Oct 2022 12:30:44 -0700
Subject: [PATCH] [BUILD] use cmake to set include path when build isn't
 triggered by setup.py (#754)

---
 CMakeLists.txt | 15 ++++++++++++---
 1 file changed, 12 insertions(+), 3 deletions(-)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index 234a10f9a..aee0045ce 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -132,9 +132,18 @@ include_directories(${LLVM_INCLUDE_DIRS})
 # Python module
 if(TRITON_BUILD_PYTHON_MODULE)
     message(STATUS "Adding Python module")
-    set(PYTHON_SRC_PATH ${CMAKE_CURRENT_SOURCE_DIR}/python/src)
-    include_directories("." ${PYTHON_SRC_PATH} ${PYTHON_INCLUDE_DIRS})
-    link_directories(${PYTHON_LINK_DIRS})
+    if (PYTHON_INCLUDE_DIRS)
+      set(PYTHON_SRC_PATH ${CMAKE_CURRENT_SOURCE_DIR}/python/src)
+      include_directories("." ${PYTHON_SRC_PATH} ${PYTHON_INCLUDE_DIRS})
+      link_directories(${PYTHON_LINK_DIRS})
+    else()
+      find_package(Python3 REQUIRED COMPONENTS Development)
+      set(PYTHON_SRC_PATH ${CMAKE_CURRENT_SOURCE_DIR}/python/src)
+      include_directories("." ${PYTHON_SRC_PATH} ${Python3_INCLUDE_DIRS})
+      link_directories(${Python3_LIBRARY_DIRS})
+      link_libraries(${Python3_LIBRARIES})
+      add_link_options(${Python3_LINK_OPTIONS})
+    endif()
     set(PYTHON_SRC ${PYTHON_SRC_PATH}/main.cc ${PYTHON_SRC_PATH}/triton.cc)
 endif()