From b459f73741a9d2131369c2f4dbacb50b6fd89a8a Mon Sep 17 00:00:00 2001
From: kthui <18255193+kthui@users.noreply.github.com>
Date: Fri, 8 Dec 2023 17:41:20 -0800
Subject: [PATCH] Do not install Python runtime on non x86

---
 CMakeLists.txt | 20 ++++++++++++--------
 1 file changed, 12 insertions(+), 8 deletions(-)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index e45e7fc..f286260 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -506,14 +506,18 @@ install(
 )
 
 if (${TRITON_PYTORCH_ENABLE_PYTHON_RUNTIME})
-  install(CODE "execute_process(COMMAND bash -c ${CMAKE_CURRENT_SOURCE_DIR}/tools/gen_pb_exec_env.sh)")
-  install(
-    FILES
-      src/model.py
-      ${CMAKE_CURRENT_BINARY_DIR}/pb_exec_env_model.py.tar.gz
-    DESTINATION
-      ${CMAKE_INSTALL_PREFIX}/backends/pytorch
-  )
+  if ((CMAKE_HOST_SYSTEM_PROCESSOR MATCHES "x86_64") OR (CMAKE_HOST_SYSTEM_PROCESSOR MATCHES "AMD64"))
+    install(CODE "execute_process(COMMAND bash -c ${CMAKE_CURRENT_SOURCE_DIR}/tools/gen_pb_exec_env.sh)")
+    install(
+      FILES
+        src/model.py
+        ${CMAKE_CURRENT_BINARY_DIR}/pb_exec_env_model.py.tar.gz
+      DESTINATION
+        ${CMAKE_INSTALL_PREFIX}/backends/pytorch
+    )
+  else()
+    message(WARNING "Skipped PyTorch Python runtime on unsupported architecture ${CMAKE_HOST_SYSTEM_PROCESSOR}")
+  endif()
 endif() # TRITON_PYTORCH_ENABLE_PYTHON_RUNTIME
 
 include(CMakePackageConfigHelpers)