Fix test: use as_tensor instead of make_tensor

2026-06-01 07:54:36 +00:00
parent 5ea71ebd78
commit d2819fc39c
1 changed files with 1 additions and 1 deletions
--- 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")