workaround metal

Author: oraluben

CMakeLists.txt

@@ -159,8 +159,10 @@ endif()
 # let libtilelang to search tvm/tvm_runtime in same dir
 if(APPLE)
   set_target_properties(tilelang PROPERTIES INSTALL_RPATH "@loader_path")
+  set_target_properties(tilelang_module PROPERTIES INSTALL_RPATH "@loader_path")
 else()
   set_target_properties(tilelang PROPERTIES INSTALL_RPATH "\$ORIGIN")
+  set_target_properties(tilelang_module PROPERTIES INSTALL_RPATH "\$ORIGIN")
 endif()
 
 install(TARGETS tvm tvm_runtime tilelang_module tilelang LIBRARY DESTINATION tilelang/lib)

tilelang/__init__.py

@@ -67,14 +67,16 @@ def _init_logger():
 # Setup tvm search path before importing tvm
 from . import libinfo
 
+from .contrib.cc import is_darwin
+
 
 def _load_tile_lang_lib():
     """Load Tile Lang lib"""
     if sys.platform.startswith("win32") and sys.version_info >= (3, 8):
         for path in libinfo.get_dll_directories():
             os.add_dll_directory(path)
     # pylint: disable=protected-access
-    lib_name = "tilelang" if tvm.base._RUNTIME_ONLY else "tilelang_module"
+    lib_name = "tilelang" if tvm.base._RUNTIME_ONLY and not is_darwin() else "tilelang_module"
     # pylint: enable=protected-access
     lib_path = libinfo.find_lib_path(lib_name, optional=False)
     return ctypes.CDLL(lib_path), lib_path