Fix: add cpasync import to test_d1_qk512

2026-05-24 14:20:01 +00:00
parent e0cd810d39
commit 04b8ca43ed
1 changed files with 1 additions and 1 deletions
--- a/tests/unit/test_d1_qk512.py
+++ b/tests/unit/test_d1_qk512.py
@@ -5,7 +5,7 @@ import cutlass.cute as cute
 import cutlass.torch as ct
 import cuda.bindings.driver as cuda
 from cutlass import BFloat16, Float32
-from cutlass.cute.nvgpu import tcgen05
+from cutlass.cute.nvgpu import cpasync, tcgen05
 from cutlass.utils import LayoutEnum
 import cutlass.utils as utils
 import cutlass.pipeline as pipeline