From 25b9a5f32d773250d18a61c3515abee2fabe82eb Mon Sep 17 00:00:00 2001 From: biondizzle Date: Mon, 1 Jun 2026 07:55:29 +0000 Subject: [PATCH] Fix test: use from_dlpack for c_tensor --- tests/unit/test_nvfp4_cutedsl_compile.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/unit/test_nvfp4_cutedsl_compile.py b/tests/unit/test_nvfp4_cutedsl_compile.py index 430a0457..e1ab9870 100644 --- a/tests/unit/test_nvfp4_cutedsl_compile.py +++ b/tests/unit/test_nvfp4_cutedsl_compile.py @@ -54,8 +54,10 @@ def test_nvfp4_cutedsl_compilation(): sfb_tensor = cutlass_torch.from_dlpack(scale_b) sfb_tensor = sfb_tensor.mark_layout_dynamic(leading_dim=cutlass_torch.get_leading_dim(scale_b)) - c_tensor = cutlass_torch.as_tensor( + c_tensor = cutlass_torch.from_dlpack( torch.empty(M, N, dtype=torch.bfloat16, device=device)) + c_tensor = c_tensor.mark_layout_dynamic(leading_dim=cutlass_torch.get_leading_dim( + torch.empty(M, N, dtype=torch.bfloat16, device=device))) print("CuTe tensors created OK")