[Kimi][MM] disable kimi_vit's dynamic torch.compile for TPU (#51196)
Signed-off-by: Linkun Chen <github@lkchen.net>
Linkun committed
Aug 7, 2026 at 17:42 UTC
7f58e8294a19eab37011c21c8da981206ca3849e
1 file changed
+1
-1
vllm/model_executor/models/kimi_k25_vit.py
+1
-1
@@ -61,7 +61,7 @@ def get_rope_shape_decorate(func):
61
62
63
@get_rope_shape_decorate
64
-@torch.compile(dynamic=True)
64
+@torch.compile(dynamic=True, disable=current_platform.simple_compile_backend == "tpu")
65
def get_rope_shape(org, interpolation_mode, shape):
66
return (
67
F.interpolate(