[Intel GPU] DeepSeek V4 2/N: Fix tvm ffi import (#26118)

This commit is contained in:
Polisetty V R K Jyothendra Varma
2026-05-24 12:59:49 -07:00
committed by GitHub
parent 44922de48a
commit fd94bd30b8
+4 -2
View File
@@ -1,9 +1,8 @@
from __future__ import annotations
from typing import Literal, NamedTuple, Optional, Union
from typing import TYPE_CHECKING, Literal, NamedTuple, Optional, Union
import torch
from tvm_ffi.module import Module
from sglang.jit_kernel.utils import (
cache_once,
@@ -14,6 +13,9 @@ from sglang.jit_kernel.utils import (
from .utils import make_name
if TYPE_CHECKING:
from tvm_ffi.module import Module
@cache_once
def _jit_compress_norm_rope_module(