Revert "[Fix] Resolve TVM Compatibility Issue by Removing Unsupported Argument" (#3165)

MasterJH5574 · web-flow · commit bbd5e9cab92a · 2025-03-10T09:29:20.000-04:00
Revert "[Fix] Resolve TVM Compatibility Issue by Removing Unsupported Argumen…" This reverts commit 7175da5.
diff --git a/python/mlc_llm/compiler_pass/dispatch_kv_cache_creation.py b/python/mlc_llm/compiler_pass/dispatch_kv_cache_creation.py
@@ -152,9 +152,6 @@ def create_tir_paged_kv_cache(
             "support_sliding_window_", relax.ShapeStructInfo([kwargs["support_sliding_window"]])
         )
 
-        # Remove the 'enable_disaggregation' argument
-        kwargs.pop("enable_disaggregation", None)
-
         with bb.function(
             name="create_tir_paged_kv_cache",
             params=[

Original file line number	Diff line number	Diff line change
`@@ -152,9 +152,6 @@ def create_tir_paged_kv_cache(`
`152`	`152`	`"support_sliding_window_", relax.ShapeStructInfo([kwargs["support_sliding_window"]])`
`153`	`153`	`)`
`154`	`154`
`155`		`- # Remove the 'enable_disaggregation' argument`
`156`		`- kwargs.pop("enable_disaggregation", None)`
`157`		`-`
`158`	`155`	`with bb.function(`
`159`	`156`	`name="create_tir_paged_kv_cache",`
`160`	`157`	`params=[`