diff --git a/python/triton/runtime/build.py b/python/triton/runtime/build.py
index 6c9f35827d73..3c8accef3c4f 100644
--- a/python/triton/runtime/build.py
+++ b/python/triton/runtime/build.py
@@ -1,5 +1,6 @@
 import contextlib
 import sys
+import platform
 import io
 import sysconfig
 import os
@@ -20,6 +21,7 @@ def quiet():
 
 def _build(name, src, srcdir, library_dirs, include_dirs, libraries):
     suffix = sysconfig.get_config_var('EXT_SUFFIX')
+    system = platform.system()
     so = os.path.join(srcdir, '{name}{suffix}'.format(name=name, suffix=suffix))
     # try to avoid setuptools if possible
     cc = os.environ.get("CC")
@@ -42,6 +44,9 @@ def _build(name, src, srcdir, library_dirs, include_dirs, libraries):
     py_include_dir = sysconfig.get_paths(scheme=scheme)["include"]
     include_dirs = include_dirs + [srcdir, py_include_dir]
     cc_cmd = [cc, src, "-O3", "-shared", "-fPIC", "-o", so]
+    # Use dynamic lookup to load Python library on Mac
+    if system == "Darwin":
+        cc_cmd += ["-undefined", "dynamic_lookup"]
     cc_cmd += [f'-l{lib}' for lib in libraries]
     cc_cmd += [f"-L{dir}" for dir in library_dirs]
     cc_cmd += [f"-I{dir}" for dir in include_dirs]
diff --git a/third_party/cpu/backend/driver.py b/third_party/cpu/backend/driver.py
index 126e41d3bd7a..ebce4229d7af 100644
--- a/third_party/cpu/backend/driver.py
+++ b/third_party/cpu/backend/driver.py
@@ -143,6 +143,7 @@ def format_of(ty):
 #include <optional>
 #include <stdio.h>
 #include <string>
+#include <memory>
 
 #define NPY_NO_DEPRECATED_API NPY_1_7_API_VERSION
 #include <Python.h>