minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

[Intel GPU] 1/N Fix tilelang import in deepseek v4 rope as optional (#25128)

Browse Source
This commit is contained in:
Polisetty V R K Jyothendra Varma
2026-05-22 18:23:18 +08:00
committed by GitHub
parent d4082eab4d
commit 80680dc3fe
1 changed files with 9 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/sglang/srt/layers/deepseek_v4_rope.py
+9 -6
View File
@@ -2,17 +2,20 @@ import math
from functools import lru_cache from functools import lru_cache
from typing import Optional from typing import Optional
import tilelang
import torch import torch
import triton import triton
import triton.language as tl import triton.language as tl
tilelang.set_log_level("WARNING") try:
import tilelang
pass_configs = { tilelang.set_log_level("WARNING")
tilelang.PassConfigKey.TL_DISABLE_WARP_SPECIALIZED: True, pass_configs = {
tilelang.PassConfigKey.TL_DISABLE_TMA_LOWER: True, tilelang.PassConfigKey.TL_DISABLE_WARP_SPECIALIZED: True,
} tilelang.PassConfigKey.TL_DISABLE_TMA_LOWER: True,
}
except ImportError:
pass
FP8 = "float8_e4m3" FP8 = "float8_e4m3"
BF16 = "bfloat16" BF16 = "bfloat16"