From d2819fc39c7c6e8a9eeef5d572422b2b2512cdf3 Mon Sep 17 00:00:00 2001 From: biondizzle Date: Mon, 1 Jun 2026 07:54:36 +0000 Subject: [PATCH] Fix test: use as_tensor instead of make_tensor --- tests/unit/test_nvfp4_cutedsl_compile.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/unit/test_nvfp4_cutedsl_compile.py b/tests/unit/test_nvfp4_cutedsl_compile.py index 823eb5bf..430a0457 100644 --- a/tests/unit/test_nvfp4_cutedsl_compile.py +++ b/tests/unit/test_nvfp4_cutedsl_compile.py @@ -54,7 +54,7 @@ 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.make_tensor( + c_tensor = cutlass_torch.as_tensor( torch.empty(M, N, dtype=torch.bfloat16, device=device)) print("CuTe tensors created OK")