[CPU] upgrade dependent torch ver to PT2.12 (#21456)

Co-authored-by: Ma Mingfei <mingfei.ma@intel.com>
This commit is contained in:
Zaili Wang
2026-06-04 11:04:11 +08:00
committed by GitHub
co-authored by Ma Mingfei
parent 29d23e198f
commit 3b7a258f63
11 changed files with 18 additions and 22 deletions
+1 -1
View File
@@ -8,7 +8,7 @@ from sglang.srt.layers.amx_utils import CPUQuantMethod
kernel = torch.ops.sgl_kernel
torch.manual_seed(1234)
torch.manual_seed(1183)
from utils import (
BLOCK_K,
+1 -1
View File
@@ -177,7 +177,7 @@ class TestROPE(CustomTestCase):
num_kv_heads: int,
):
set_global_server_args_for_scheduler(ServerArgs(model_path="dummy"))
torch.manual_seed(100)
torch.manual_seed(1234)
rope_ref = RotaryEmbedding(
head_size,
rotary_dim,
+1 -3
View File
@@ -15,13 +15,11 @@ from sglang.test.test_utils import CustomTestCase
register_cpu_ci(est_time=10, suite="base-b-test-cpu")
torch.manual_seed(1234)
# This is used by the Deepseek-V2 model
class TestGroupedTopK(CustomTestCase):
def _run_single_test(self, M, E, G, topk, topk_group, renormalize, dtype):
torch.manual_seed(1234)
torch.manual_seed(12)
# expand gating_output by M, otherwise bfloat16 fall into same value aftering truncating
hidden_states = torch.randn(M, 100, dtype=dtype)
+1 -1
View File
@@ -9,7 +9,7 @@ from sglang.srt.layers.amx_utils import CPUQuantMethod
kernel = torch.ops.sgl_kernel
torch.manual_seed(128)
torch.manual_seed(1183)
from utils import (
BLOCK_K,
+1 -1
View File
@@ -174,7 +174,7 @@ class TestROPE(CustomTestCase):
num_kv_heads: int,
):
set_global_server_args_for_scheduler(ServerArgs(model_path="dummy"))
torch.manual_seed(100)
torch.manual_seed(1234)
rope_ref = RotaryEmbedding(
head_size,
rotary_dim,
+1 -3
View File
@@ -10,13 +10,11 @@ from sglang.srt.layers.moe.topk import grouped_topk_gpu as native_grouped_topk
from sglang.srt.models.llama4 import Llama4MoE
from sglang.test.test_utils import CustomTestCase
torch.manual_seed(1234)
# This is used by the Deepseek-V2 model
class TestGroupedTopK(CustomTestCase):
def _run_single_test(self, M, E, G, topk, topk_group, renormalize, dtype):
torch.manual_seed(1234)
torch.manual_seed(12)
# expand gating_output by M, otherwise bfloat16 fall into same value aftering truncating
hidden_states = torch.randn(M, 100, dtype=dtype)