ci fix 4

Author: oraluben

CMakeLists.txt

@@ -20,6 +20,7 @@ set(USE_ROCM OFF)
 set(USE_METAL OFF)
 set(PREBUILD_CYTHON ON)
 set(STATIC_LIB OFF)
+set(RUNTIME_ONLY TRUE)
 # Configs end
 
 find_package(TVM REQUIRED)
@@ -78,6 +79,8 @@ else()
 endif()
 
 if(USE_METAL)
+  set(RUNTIME_ONLY FALSE)
+
   file(GLOB TILE_LANG_METAL_SRCS
     src/target/rt_mod_metal.cc
   )
@@ -131,9 +134,12 @@ add_library(tilelang_objs OBJECT ${TILE_LANG_SRCS})
 target_include_directories(tilelang_objs PRIVATE ${TILE_LANG_INCLUDES})
 
 add_library(tilelang SHARED $<TARGET_OBJECTS:tilelang_objs>)
-add_library(tilelang_module SHARED $<TARGET_OBJECTS:tilelang_objs>)
-target_link_libraries(tilelang PUBLIC tvm_runtime)
-target_link_libraries(tilelang_module PUBLIC tvm)
+target_link_libraries(tilelang PUBLIC tvm)
+if(RUNTIME_ONLY)
+  target_link_libraries(tilelang PUBLIC tvm_runtime)
+else()
+  target_link_libraries(tilelang PUBLIC tvm)
+endif()
 
 if(PREBUILD_CYTHON)
   find_package(Python REQUIRED COMPONENTS Interpreter Development.Module ${SKBUILD_SABI_COMPONENT})
@@ -159,13 +165,11 @@ 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)
+install(TARGETS tvm tvm_runtime tilelang LIBRARY DESTINATION tilelang/lib)
 
 # Copy tvm cython ext for wheels
 # TODO: not necessary for editable builds

tilelang/__init__.py

@@ -76,7 +76,7 @@ def _load_tile_lang_lib():
         for path in libinfo.get_dll_directories():
             os.add_dll_directory(path)
     # pylint: disable=protected-access
-    lib_name = "tilelang" if tvm.base._RUNTIME_ONLY and not is_darwin() else "tilelang_module"
+    lib_name = "tilelang"
     # pylint: enable=protected-access
     lib_path = libinfo.find_lib_path(lib_name, optional=False)
     return ctypes.CDLL(lib_path), lib_path