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

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
+9 -6
View File
@@ -2,17 +2,20 @@ import math
from functools import lru_cache
from typing import Optional
import tilelang
import torch
import triton
import triton.language as tl
tilelang.set_log_level("WARNING")
try:
import tilelang
pass_configs = {
tilelang.PassConfigKey.TL_DISABLE_WARP_SPECIALIZED: True,
tilelang.PassConfigKey.TL_DISABLE_TMA_LOWER: True,
}
tilelang.set_log_level("WARNING")
pass_configs = {
tilelang.PassConfigKey.TL_DISABLE_WARP_SPECIALIZED: True,
tilelang.PassConfigKey.TL_DISABLE_TMA_LOWER: True,
}
except ImportError:
pass
FP8 = "float8_e4m3"
BF16 = "bfloat16"