From e9f476b6dcac11f34c41433505ba609b3fd8d11f Mon Sep 17 00:00:00 2001 From: biondizzle Date: Mon, 25 May 2026 17:28:43 +0000 Subject: [PATCH] =?UTF-8?q?fix=20typo:=20from=5Fdlset=20=E2=86=92=20from?= =?UTF-8?q?=5Fdlpack?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- tests/unit/test_d3_swa_mask.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/unit/test_d3_swa_mask.py b/tests/unit/test_d3_swa_mask.py index c066d289..df4ad64f 100644 --- a/tests/unit/test_d3_swa_mask.py +++ b/tests/unit/test_d3_swa_mask.py @@ -58,7 +58,7 @@ def _run_fmha(q_3d, k_3d, v, m, s_k, hd, use_smem_p=False): c_tile.zero_(); lse_tensor.zero_() mV = ct.from_dlpack(v_tile).mark_layout_dynamic(leading_dim=ct.get_leading_dim(v_tile)) mC = ct.from_dlpack(c_tile).mark_layout_dynamic(leading_dim=ct.get_leading_dim(c_tile)) - mLSE = ct.from_dlset(lse_tensor).mark_layout_dynamic(leading_dim=ct.get_leading_dim(lse_tensor)) + mLSE = ct.from_dlpack(lse_tensor).mark_layout_dynamic(leading_dim=ct.get_leading_dim(lse_tensor)) compiled(mQ, mK, mV, mC, stream, mLSE) o_unnorm[:, pv*pv_n_tile:(pv+1)*pv_n_tile] = c_tile[:,:,0].float()