--- setup.py_orig	2025-05-15 22:12:39.672759723 -0500
+++ setup.py	2025-05-15 22:14:28.447271573 -0500
@@ -253,6 +253,7 @@
 import shutil
 import subprocess
 import sysconfig
+import platform
 import time
 from collections import defaultdict
 
@@ -1031,6 +1032,8 @@
             return ["-Wl,-rpath,@loader_path/" + path]
         elif IS_WINDOWS:
             return []
+        elif platform.system() == 'AIX':
+            return ["-Wl,-blibpath:/opt/freeware/lib/pthread:/opt/freeware/lib64/:/opt/freeware/lib:/opt/freeware/lib64/python3.12/site-packages/torch/lib/:/usr/lib:/lib" + path]
         else:
             return ["-Wl,-rpath,$ORIGIN/" + path]
 
@@ -1390,6 +1393,7 @@
                 "lib/libtorch_python.so",
                 "lib/libtorch_python.dylib",
                 "lib/libtorch_python.dll",
+                "lib/libtorch_python.a",
             ]
         )
     if not BUILD_PYTHON_ONLY:
@@ -1399,6 +1403,7 @@
                 "lib/*.dylib*",
                 "lib/*.dll",
                 "lib/*.lib",
+                "lib/*.a",
             ]
         )
         aotriton_image_path = os.path.join(lib_path, "aotriton.images")