[CPU] Explicitly import sgl_kernel in CPU kernel tests (#35119)

This commit is contained in:
Ma Mingfei
2026-08-18 10:20:28 +08:00
committed by GitHub
parent d55f1c28e2
commit 7f51e6bba0
16 changed files with 16 additions and 0 deletions
+1
View File
@@ -1,5 +1,6 @@
import unittest
import sgl_kernel # noqa: F401
import torch
from torch.nn.functional import scaled_dot_product_attention