I’m currently working on refactoring our project of the methodology we discussed in this thread, using TVM core infrastructure by utilizing tvm with include dependencies and link tvm with shared libraries.
Example of a CMakeList.txt that works with tvm.
cmake_minimum_required(VERSION 3.21)
project(TileLang C CXX)
set(CMAKE_CXX_STANDARD 17)
# Define TVM root directory
set(TVM_ROOT ${PROJECT_SOURCE_DIR}/3rdparty/tvm)
# Include directories
include_directories(
${PROJECT_SOURCE_DIR}/include
)
# Source files for the project
file(GLOB_RECURSE TileLang_SOURCES
${PROJECT_SOURCE_DIR}/src/transform/*.cpp
${PROJECT_SOURCE_DIR}/src/op/*.cc
${PROJECT_SOURCE_DIR}/src/codegen/*.cc
)
# Create shared library
add_library(TileLang SHARED ${TileLang_SOURCES})
I think the key part of this pipeline is, ensuring that the tvm based implementation allows developers write their own passes(from the cpp side), I’m not sure how we can still bind our own cpp transformations and op define to python with TVM FFI. Do we have any example projects or guidelines for this? I’ll continue exploring to achieve a cleaner design.