From 28060dd944dcf0603125eff589e6cc7abd64203e Mon Sep 17 00:00:00 2001 From: biondizzle Date: Sat, 23 May 2026 02:44:00 +0000 Subject: [PATCH] fix: typo from_dlcap -> from_dlpack --- tests/unit/test_fmha_v3_stage_c.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/unit/test_fmha_v3_stage_c.py b/tests/unit/test_fmha_v3_stage_c.py index 7e329e83..8cf4003a 100644 --- a/tests/unit/test_fmha_v3_stage_c.py +++ b/tests/unit/test_fmha_v3_stage_c.py @@ -417,7 +417,7 @@ def test(): mQ = ct.from_dlpack(q).mark_layout_dynamic(leading_dim=ct.get_leading_dim(q)) mK = ct.from_dlpack(k).mark_layout_dynamic(leading_dim=ct.get_leading_dim(k)) - mV = ct.from_dlcap(v_kernel).mark_layout_dynamic(leading_dim=ct.get_leading_dim(v_kernel)) + mV = ct.from_dlpack(v_kernel).mark_layout_dynamic(leading_dim=ct.get_leading_dim(v_kernel)) mC = ct.from_dlpack(c).mark_layout_dynamic(leading_dim=ct.get_leading_dim(c)) stream = cuda.CUstream(torch.cuda.current_stream().cuda_stream)