From c46526a9f1282bf1f8064c6241faeb0f4dec4dd1 Mon Sep 17 00:00:00 2001 From: Siyuan Feng <25500082+Hzfengsy@users.noreply.github.com> Date: Wed, 9 Jul 2025 11:34:32 +0800 Subject: [PATCH] Phase out StackVM runtime support Previous PRs have removed the stack vm support, this PR is to clean up the remaining references to it, and replace the default target from stackvm to c backend. --- CMakeLists.txt | 1 - cmake/modules/LibInfo.cmake | 1 - python/tvm/ffi/cython/device.pxi | 1 - python/tvm/relax/vm_build.py | 2 +- python/tvm/tir/build.py | 2 +- src/support/libinfo.cc | 1 - 6 files changed, 2 insertions(+), 6 deletions(-) diff --git a/CMakeLists.txt b/CMakeLists.txt index b90fc43e056c..a3e62bc26750 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -56,7 +56,6 @@ tvm_option(USE_RPC "Build with RPC" ON) tvm_option(USE_THREADS "Build with thread support" ON) tvm_option(USE_LLVM "Build with LLVM, can be set to specific llvm-config path" OFF) tvm_option(USE_MLIR "Build with MLIR support" OFF) -tvm_option(USE_STACKVM_RUNTIME "Include stackvm into the runtime" OFF) tvm_option(USE_OPENMP "Build with OpenMP thread pool implementation" OFF) tvm_option(TVM_DEBUG_WITH_ABI_CHANGE "Enable debug code that may cause ABI changes" OFF) tvm_option(TVM_LOG_BEFORE_THROW "Whether log before throw, for debugging purposes" OFF) diff --git a/cmake/modules/LibInfo.cmake b/cmake/modules/LibInfo.cmake index 14c63f8aa353..73d789e9fa94 100644 --- a/cmake/modules/LibInfo.cmake +++ b/cmake/modules/LibInfo.cmake @@ -112,7 +112,6 @@ function(add_lib_info src_file) TVM_INFO_USE_RUST_EXT="${USE_RUST_EXT}" TVM_INFO_USE_SORT="${USE_SORT}" TVM_INFO_USE_SPIRV_KHR_INTEGER_DOT_PRODUCT="${USE_SPIRV_KHR_INTEGER_DOT_PRODUCT}" - TVM_INFO_USE_STACKVM_RUNTIME="${USE_STACKVM_RUNTIME}" TVM_INFO_USE_TENSORFLOW_PATH="${USE_TENSORFLOW_PATH}" TVM_INFO_USE_TENSORRT_CODEGEN="${USE_TENSORRT_CODEGEN}" TVM_INFO_USE_TENSORRT_RUNTIME="${USE_TENSORRT_RUNTIME}" diff --git a/python/tvm/ffi/cython/device.pxi b/python/tvm/ffi/cython/device.pxi index ed7d22e9a6ab..90d641c44ffa 100644 --- a/python/tvm/ffi/cython/device.pxi +++ b/python/tvm/ffi/cython/device.pxi @@ -78,7 +78,6 @@ cdef class Device: DEVICE_NAME_TO_TYPE = { "llvm": kDLCPU, - "stackvm": kDLCPU, "cpu": kDLCPU, "c": kDLCPU, "test": kDLCPU, diff --git a/python/tvm/relax/vm_build.py b/python/tvm/relax/vm_build.py index 3332df6c8735..f31927e2f1f9 100644 --- a/python/tvm/relax/vm_build.py +++ b/python/tvm/relax/vm_build.py @@ -192,7 +192,7 @@ def build( to setup the dimensions and parameters correctly. host is used to specify the host side codegen target. By default, llvm is used if it is enabled, - otherwise a stackvm interpreter is used. + otherwise a c backend is used. params: Optional[Dict[str, list]] Parameters for the input IRModule that will be bound. diff --git a/python/tvm/tir/build.py b/python/tvm/tir/build.py index 14bc189b9f6d..1948ca193f28 100644 --- a/python/tvm/tir/build.py +++ b/python/tvm/tir/build.py @@ -150,7 +150,7 @@ def build( target = Target.canon_target(target) # Step 2: Determine the host target - target_host = "llvm" if tvm.runtime.enabled("llvm") else "stackvm" + target_host = "llvm" if tvm.runtime.enabled("llvm") else "c" if target is not None: if target.host is not None: target_host = target.host diff --git a/src/support/libinfo.cc b/src/support/libinfo.cc index 6723cf06397d..e31fddbcb058 100644 --- a/src/support/libinfo.cc +++ b/src/support/libinfo.cc @@ -343,7 +343,6 @@ TVM_DLL ffi::Map GetLibInfo() { {"USE_RUST_EXT", TVM_INFO_USE_RUST_EXT}, {"USE_SORT", TVM_INFO_USE_SORT}, {"USE_SPIRV_KHR_INTEGER_DOT_PRODUCT", TVM_INFO_USE_SPIRV_KHR_INTEGER_DOT_PRODUCT}, - {"USE_STACKVM_RUNTIME", TVM_INFO_USE_STACKVM_RUNTIME}, {"USE_TENSORFLOW_PATH", TVM_INFO_USE_TENSORFLOW_PATH}, {"USE_TENSORRT_CODEGEN", TVM_INFO_USE_TENSORRT_CODEGEN}, {"USE_TENSORRT_RUNTIME", TVM_INFO_USE_TENSORRT_RUNTIME},