add seed in CPU UTs to avoid flaky failure (#7333)

This commit is contained in:
Chunyuan WU
2025-06-19 10:12:14 +08:00
committed by GitHub
parent 20a503c7d1
commit 9179ea1595
10 changed files with 20 additions and 0 deletions

View File

@@ -8,6 +8,8 @@ import torch
kernel = torch.ops.sgl_kernel
torch.manual_seed(0)
from utils import (
BLOCK_K,
BLOCK_N,